在Numpy中,可以使用索引数组来索引多维数组,以产生所需的结果。在这种情况下,我们可以使用以下方法来实现:
import numpy as np
array_3d = np.random.random((3, 4, 5))
index_array = np.array([0, 2, 1])
result_array = array_3d[index_array, :, :]
这样,我们就可以通过使用大小为B的索引数组来索引大小为B x N x M的多维数组,得到一个大小为B x M的结果数组。
注意:以上代码示例中,B、N、M均为正整数,代表相应的维度大小。另外,示例中使用了随机数组来演示,实际情况中可以根据具体需求来创建和使用数组。
关于Numpy的详细信息和更多功能,您可以查阅腾讯云官方文档中的Numpy相关部分:Numpy - 腾讯云官方文档
领取专属 10元无门槛券
手把手带您无忧上云