LazyConvTranspose3d (original) (raw)

class torch.nn.LazyConvTranspose3d(out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1, bias=True, dilation=1, padding_mode='zeros', device=None, dtype=None)[source]#

A torch.nn.ConvTranspose3d module with lazy initialization of the in_channels argument.

The in_channels argument of the ConvTranspose3d is inferred from the input.size(1). The attributes that will be lazily initialized are weight and bias.

Check the torch.nn.modules.lazy.LazyModuleMixin for further documentation on lazy modules and their limitations.

Parameters

cls_to_become[source]#

alias of ConvTranspose3d