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#
1D transposed convolution over an input with several channels
- Parameters:
input (array) – Input array of shape
(N, L, C_in).weight (array) – Weight array of shape
(C_out, K, C_in).stride (int, optional) – Kernel stride. Default:
1.padding (int, optional) – Input padding. Default:
0.dilation (int, optional) – Kernel dilation. Default:
1.output_padding (int, optional) – Output padding. Default:
0.groups (int, optional) – Input feature groups. Default:
1.
- Returns:
The convolved array.
- Return type: