mlx.core.conv_transpose2d

mlx.core.conv_transpose2d#

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

对具有多个通道的输入执行二维转置卷积

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

参数：

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

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

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

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

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

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

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

返回：

卷积后的数组。

返回类型：

array

目录