LazyConv3d

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

一個 torch.nn.Conv3d 模組，對 in_channels 引數進行惰性初始化。

Conv3d 模組的 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.Conv3d 和 torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

Conv3d 的別名