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)

一个带有in_channels参数惰性初始化的torch.nn.Conv2d模块。

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

查看 torch.nn.modules.lazy.LazyModuleMixin 以获取关于惰性模块及其限制的更多文档。

Parameters

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

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

• 步幅 (整数 或 元组, 可选) – 卷积的步幅。默认值：1

• 填充 (int 或 tuple, 可选) – 在输入的两边添加的零填充。默认值：0

• padding_mode (字符串, 可选) – 'zeros', 'reflect', 'replicate' 或 'circular'。默认值: 'zeros'

• 膨胀 (int 或 tuple, 可选) – 核元素之间的间距。默认值：1

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

• 偏置 (布尔值, 可选) – 如果True，则在输出中添加一个可学习的偏置。默认值：True

另请参阅

torch.nn.Conv2d 和 torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

的别名 Conv2d