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

对具有多个通道的输入进行一维卷积

参数:

• 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。

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

返回值:

卷积后的数组。

返回值类型:

array