LazyConv3d

classtorch.nn.LazyConv3d(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.Conv3d模块。

in_channels 参数是从 input.size(1) 推断出来的，用于 Conv3d。将被惰性初始化的属性是 weight 和 bias。

查阅torch.nn.modules.lazy.LazyModuleMixin以了解更多关于延迟模块及其限制的文档。

参数

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

• kernel_size (int 或 tuple) – 卷积核的尺寸

• stride (int 或 元组, 可选) – 卷积的步长。默认值：1

• padding (int 或 元组, 可选) – 输入两侧添加的零填充。默认值：0

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

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

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

• padding_mode (str, optional) – 'zeros', 'reflect', 'replicate' 或 'circular'。默认值：'zeros'

参见

torch.nn.Conv3d 和 torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

别名：Conv3d