mlx.core.atleast_3d

目录

mlx.core.atleast_3d#

atleast_3d(*arys: array, stream: None | Stream | Device = None) array | list[array]#

将所有数组转换为至少三维。

参数:
  • *arys – 输入数组。

  • stream (Union[None, Stream, Device], 可选) – 执行操作所用的流。

返回值:

至少三维的数组或数组列表。

返回类型:

arraylist(array)