LazyConv1d

class torch.nn.LazyConv1d(out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', device=None, dtype=None)

一个 torch.nn.Conv1d 模块，其 in_channels 参数会延迟初始化。

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

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

参数

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

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

• stride (int 或 tuple, 可选) – 卷积的步长。默认为 1

• padding (int 或 tuple, 可选) – 添加到输入两侧的零填充。默认为 0

• dilation (int 或 tuple, 可选) – 卷积核元素之间的间距。默认为 1

• groups (int, 可选) – 从输入通道到输出通道的分组连接数。默认为 1

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

• padding_mode (str, 可选) – 'zeros'、'reflect'、'replicate' 或 'circular'。默认为 'zeros'

另请参阅

torch.nn.Conv1d 和 torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

Conv1d 的别名