LazyConvTranspose2d

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

一個具有 torch.nn.ConvTranspose2d 的模組，它對 in_channels 引數進行了延遲初始化。

ConvTranspose2d 的 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，可選) – 在輸入的每個維度兩側會新增 dilation * (kernel_size - 1) - padding 的零填充。預設值：0

• output_padding (int 或 tuple，可選) – 在輸出形狀的每個維度的一側新增的額外尺寸。預設值：0

• groups (int, 可選) – 從輸入通道到輸出通道的阻塞連線數。預設為：1

• bias (bool, 可選) – 如果為 True，則向輸出新增可學習的偏置。預設為：True

• dilation (int 或 tuple, 可選) – 核元素之間的間距。預設為：1

另請參閱

torch.nn.ConvTranspose2d 和 torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

別名：ConvTranspose2d