mlx.core.expand_dims

expand_dims(a: array, /, axis: int | Sequence[int], *, stream: None | Stream |Device = None) → array

在给定的轴上添加一个大小为 1 的维度。

参数：

• a (array) – 输入数组。

• axes (int 或 tuple(int)) – 插入维度的索引。

返回：

插入维度后的数组。

返回类型：

array