LazyConv2d(out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', device=None, dtype=None)¶
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
padding_mode (string, optional) –
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: