mlx.nn.ConvTranspose3d

mlx.nn.ConvTranspose3d#

class ConvTranspose3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, output_padding: int | tuple = 0, bias: bool = True)#

对多通道输入图像应用三维转置卷积。

通道预期在最后一维，即输入形状应为 NDHWC，其中

  • N 是批量维度

  • D 是输入图像深度

  • H 是输入图像高度

  • W 是输入图像宽度

  • C 是输入通道数

参数:

  • in_channels (int) – 输入通道数。

  • out_channels (int) – 输出通道数。

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

  • stride (int 或 tuple, 可选) – 应用滤波器时的步长大小。默认值: 1。

  • padding (int 或 tuple, 可选) – 输入周围补0的量。默认值: 0。

  • dilation (int 或 tuple, 可选) – 卷积的扩张（dilation）。

  • output_padding (int 或 tuple, 可选) – 输出形状一侧额外增加的大小。默认值: 0。

  • bias (bool, 可选) – 如果为 True，则在输出中添加可学习的偏置（bias）。默认值: True

方法

目录