LazyConvTranspose2d

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

一个具有延迟初始化 in_channels 参数的 torch.nn.ConvTranspose2d 模块。

ConvTranspose2d 的 in_channels 参数是从 input.size(1) 推断出来的。将延迟初始化的属性是 weight 和 bias。

有关延迟模块及其局限性的更多文档，请查看 torch.nn.modules.lazy.LazyModuleMixin 。

参数

• out_channels (int) – 卷积产生的通道数

• kernel_size (int 或 tuple) – 卷积核的大小

• stride (int 或 tuple, optional) – 卷积的步长。默认为：1

• padding (int 或 tuple, optional) – dilation * (kernel_size - 1) - padding 将会填充零到输入每个维度的两边。默认为：0

• output_padding (int 或 tuple, optional) – 在输出形状的每个维度的一边添加的附加大小。默认为：0

• groups (int, optional) – 从输入通道到输出通道的阻塞连接数。默认为：1

• bias (bool, optional) – 如果为 True，则在输出中添加可学习的偏置。默认为：True

• dilation (int 或 tuple, optional) – 核元素之间的间距。默认为：1

另请参阅

torch.nn.ConvTranspose2d 和 torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

alias of ConvTranspose2d