快捷方式

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)[source]

一个 torch.nn.ConvTranspose2d 模块,它对 in_channels 参数进行延迟初始化。

in_channels 参数 ConvTranspose2d 来自 input.size(1) 推断。将延迟初始化的属性是 weightbias

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

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

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

  • stride (int or tuple, 可选) – 卷积的步长。默认值:1

  • padding (int or tuple, 可选) – dilation * (kernel_size - 1) - padding 零填充将添加到输入的每个维度的两侧。默认值:0

  • output_padding (int or tuple, 可选) – 添加到输出形状中每个维度一侧的额外大小。默认值:0

  • groups (int, 可选) – 从输入通道到输出通道的阻塞连接数。默认值:1

  • bias (bool, 可选) – 如果为 True,则向输出添加可学习的偏差。默认值:True

  • dilation (int or tuple, 可选) – 内核元素之间的间距。默认值:1

cls_to_become

ConvTranspose2d 的别名

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获取答案

查看资源