cupy.atleast_3d#
- cupy.atleast_3d(*arys)[source]#
将数组转换为维度 >= 3 的数组。
如果输入数组的维度小于三，则此函数插入新的轴使其具有三个维度。新轴的位置如下：
如果其形状是
()，则输出形状是(1, 1, 1)。如果其形状是
(N,)，则输出形状是(1, N, 1)。如果其形状是
(M, N)，则输出形状是(M, N, 1)。否则，输出就是输入数组本身。
- 参数:
arys (tuple of arrays) – 要转换的数组。所有参数必须是
cupy.ndarray对象。- 返回:
如果只有一个输入，则返回其转换后的版本。否则，返回一个转换后数组的列表。