快捷方式

LazyConv2d

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)[源代码]

一个 torch.nn.Conv2d 模块,其中 in_channels 参数的初始化是延迟的。

input.size(1) 推断出 Conv2din_channels 参数。将延迟初始化的属性为 weightbias

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

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

  • kernel_size (inttuple) – 卷积核的大小

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

  • padding (inttuple, 可选) – 添加到输入两侧的零填充。默认值:0

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

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

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

  • padding_mode (str, 可选) – 'zeros''reflect''replicate''circular'。默认值:'zeros'

cls_to_become

Conv2d 的别名

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源