torch.nn.modules.conv.Conv2d Class Reference

Inheritance diagram for torch.nn.modules.conv.Conv2d:

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:

• torch/nn/modules/conv.py