LazyConv3d ¬
- class torch.nn.LazyConv3d(输出通道数, 卷积核大小, 步长=1, 填充=0, 扩展率=1, 分组=1, 偏置=True, 填充模式='zeros', 设备=None, 数据类型=None)[源代码][源代码] ¬
一个具有懒加载
in_channels
参数的torch.nn.Conv3d
模块。由
Conv3d
推断出的in_channels
参数。将延迟初始化的属性是权重和偏置。检查
torch.nn.modules.lazy.LazyModuleMixin
以获取关于懒加载模块及其限制的进一步文档。- 参数:
out_channels (int) – Number of channels produced by the convolution
卷积核大小(整数或元组)- 卷积核的大小
stride(int 或 tuple,可选)- 卷积的步长。默认:1
填充(int 或 tuple,可选)- 添加到输入两边的零填充。默认:0
扩展 (int 或 tuple,可选) – 卷积核元素之间的间距。默认:1
groups(整数,可选)- 从输入通道到输出通道的阻塞连接数量。默认:1
bias(布尔值,可选)- 如果
True
,则向输出添加可学习的偏置。默认:True
padding_mode(str,可选)-
'zeros'
,'reflect'
,'replicate'
或'circular'
。默认:'zeros'
参见
torch.nn.Conv3d
和torch.nn.modules.lazy.LazyModuleMixin