- class torch.nn.LazyConv2d(out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', device=None, dtype=None)[source]¶
torch.nn.Conv2dmodule with lazy initialization of the
in_channelsargument of the
Conv2dthat is inferred from the
input.size(1). The attributes that will be lazily initialized are weight and bias.
torch.nn.modules.lazy.LazyModuleMixinfor further documentation on lazy modules and their limitations.
out_channels (int) – Number of channels produced by the convolution
stride (int or tuple, optional) – Stride of the convolution. Default: 1
padding (int or tuple, optional) – Zero-padding added to both sides of the input. Default: 0
padding_mode (str, optional) –
dilation (int or tuple, optional) – Spacing between kernel elements. Default: 1
groups (int, optional) – Number of blocked connections from input channels to output channels. Default: 1
bias (bool, optional) – If
True, adds a learnable bias to the output. Default: