cupy.atleast_2d

cupy.atleast_2d(*arys)

将数组转换为维度 >= 2 的数组。

如果输入数组的维度小于 2，则此函数会在维度的头部插入新的轴以使其具有两个维度。

参数:

arys (tuple 数组) – 要转换的数组。所有参数必须是 cupy.ndarray 对象。

返回:

如果只有一个输入，则返回其转换后的版本。否则，返回转换后的数组列表。

另请参阅

numpy.atleast_2d()