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
- 参数:
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.
- 返回:
The convolved array.
- 返回类型: