cupy.ix_

cupy.ix_(*args)

从多个序列构建开放网格。

此函数接受 N 个一维序列，并返回 N 个各自具有 N 维的输出，其中除一个维度外，所有维度的形状都为 1，并且具有非单位形状值的维度会循环遍历所有 N 个维度。

使用 ix_ 可以快速构建索引数组，用于索引交叉乘积。 a[cupy.ix_([1,3],[2,5])] 返回数组 [[a[1,2] a[1,5]], [a[3,2] a[3,5]]]。

参数:

*args – 一维序列

返回:

N 个数组，每个数组具有 N 维，其中 N 是输入序列的数量。这些数组共同构成一个开放网格。

返回类型:

tuple of ndarrays

示例

>>> a = cupy.arange(10).reshape(2, 5)
>>> a
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])
>>> ixgrid = cupy.ix_([0,1], [2,4])
>>> ixgrid
(array([[0],
       [1]]), array([[2, 4]]))

警告

此函数可能会同步设备。

另请参阅

numpy.ix_()