mlx.core.conv_transpose3d

mlx.core.conv_transpose3d#

conv_transpose3d(input: array, weight: array, /, stride: int | Tuple[int, int, int] = 1, padding: int | Tuple[int, int, int] = 0, dilation: int | Tuple[int, int, int] = 1, output_padding: int | Tuple[int, int, int] = 0, groups: int = 1, *, stream: None | Stream | Device = None) → array#

具有多个通道的输入的 3D 转置卷积。

注意：目前仅支持默认的 groups=1。

参数:

  • input (array) – 形状为 (N, D, H, W, C_in) 的输入数组。

  • weight (array) – 形状为 (C_out, KD, KH, KW, C_in) 的权重数组。

  • stride (int 或 tuple(int), 可选) – 大小为 3 的元组，包含内核步长。如果只指定一个数字，则所有空间维度都使用相同的步长。默认值：1。

  • padding (int 或 tuple(int), 可选) – 大小为 3 的元组，包含对称输入填充。如果只指定一个数字，则所有空间维度都使用相同的填充。默认值：0。

  • dilation (int 或 tuple(int), 可选) – 大小为 3 的元组，包含内核扩张（膨胀）。如果只指定一个数字，则所有空间维度都使用相同的扩张。默认值：1

  • output_padding (int 或 tuple(int), 可选) – 大小为 3 的元组，包含输出填充。如果只指定一个数字，则所有空间维度都使用相同的输出填充。默认值：0。

  • groups (int, 可选) – 输入特征组。默认值：1。

返回值:

卷积后的数组。

返回类型:

array

目录