mlx.core.conv_transpose1d#
- conv_transpose1d(input: array, weight: array, /, stride: int = 1, padding: int = 0, dilation: int = 1, 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, H, C_in).weight (array) – Weight array of shape
(C_out, H, C_in).stride (int, optional) – Kernel stride. Default:
1.padding (int, optional) – Input padding. Default:
0.dilation (int, optional) – Kernel dilation. Default:
1.groups (int, optional) – Input feature groups. Default:
1.
- Returns:
The convolved array.
- Return type: