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)
推断出Conv2d
的in_channels
参数。将延迟初始化的属性为 weight 和 bias。有关延迟模块及其限制的更多文档,请查看
torch.nn.modules.lazy.LazyModuleMixin
。