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)[source][source]
一個 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'
-
cls_to_become[source]
torch.nn.modules.conv.Conv1d 的別名