Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.modules.conv.Conv2d Class Reference
Inheritance diagram for torch.nn.modules.conv.Conv2d:
torch.nn.modules.conv._ConvNd

Public Member Functions

def __init__ (self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')
 
def forward (self, input)
 
- Public Member Functions inherited from torch.nn.modules.conv._ConvNd
def __init__ (self, in_channels, out_channels, kernel_size, stride, padding, dilation, transposed, output_padding, groups, bias, padding_mode)
 
def reset_parameters (self)
 
def extra_repr (self)
 

Public Attributes

 padding_mode
 
- Public Attributes inherited from torch.nn.modules.conv._ConvNd
 in_channels
 
 out_channels
 
 kernel_size
 
 stride
 
 padding
 
 dilation
 
 transposed
 
 output_padding
 
 groups
 
 padding_mode
 
 weight
 
 bias
 

Detailed Description

Definition at line 201 of file conv.py.


The documentation for this class was generated from the following file: