x
我有一个形状为的数组(308, 308, 308)
。v
我有一个shape的整数索引数组(10, 308, 308, 3)
。我想索引x
以创建一个y
形状矩阵,(10, 308, 308)
其(i, j, k)
-entry 是x[tuple(v[i, j, k])]
. 有没有一种快速的方法来实现这一点?
问问题
27 次
x
我有一个形状为的数组(308, 308, 308)
。v
我有一个shape的整数索引数组(10, 308, 308, 3)
。我想索引x
以创建一个y
形状矩阵,(10, 308, 308)
其(i, j, k)
-entry 是x[tuple(v[i, j, k])]
. 有没有一种快速的方法来实现这一点?