syft.frameworks.torch.nn.conv

Module Contents

class syft.frameworks.torch.nn.conv.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=False, padding_mode='zeros')

Bases: torch.nn.Module

This class is the beginning of an exact python port of the torch.nn.Conv2d module. Because PySyft cannot hook into layers which are implemented in C++, our special functionalities (such as encrypted computation) do not work with torch.nn.Conv2d and so we must have python ports available for all layer types which we seek to use.

Note that this module has been tested to ensure that it outputs the exact output values that the main module outputs in the same order that the main module does.

However, there is often some rounding error of unknown origin, usually less than 1e-6 in magnitude.

This module has not yet been tested with GPUs but should work out of the box.

forward(self, data)