mlx.core.conv1d#
- conv1d(input: array, weight: array, /, stride: int = 1, padding: int = 0, dilation: int = 1, groups: int = 1, *, stream: None | Stream | Device = None) array#
1D 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.groups (int, optional) -- Input feature groups. Default:
1.
- 回傳:
The convolved array.
- 回傳型別: