mlx.core.repeat

repeat(array: array, repeats: int, axis: int | None = None, *, stream: None | Stream | Device = None) → array

沿指定轴重复数组。

参数：

• array (array) – 输入数组。

• repeats (int) – 每个元素的重复次数。

• axis (int, 可选) – 沿其重复数组的轴。如果未指定，则使用输入的展平数组并沿轴 0 重复。

• stream (Stream, 可选) – 流或设备。默认为 None。

返回值：

重复后的结果数组。

返回类型：

array