mlx.core.conv_transpose1d

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

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

参数：

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

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

• stride (int, optional) – 核步长。默认值：1。

• padding (int, optional) – 输入填充。默认值：0。

• dilation (int, optional) – 核膨胀。默认值：1。

• output_padding (int, optional) – 输出填充。默认值：0。

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

返回值：

卷积结果数组。

返回类型：

array