我有个Numpy张量器 X = np.arange(64).reshape((4,4,4)) 我想要获取这个张量的第一维的2,3,4项,你可以这样做, Y = X[[1,2,3],:,:] 这是不是比显式地写出索引1,2,3更简单的方式呢?我尝试了像1,:这样的东西,它给我一个错误。 上下文:对于我的实际应用程序,张量的形状类似于(30000,100,100)。我想抓住这个张量的最后一个(10000,100,100)到(30000,1
在我代码中,我让for循环遍历多维numpy数组,并使用在每次迭代中获得的子数组执行一些操作。它看起来像这样 #do stuff using sub
现在使用sub完成的工作是完全矢量化的,所以它应该是有效的。另一方面,这个循环迭代大约~10^5次,这是一个瓶颈。我不太愿意这样做,因为do stuff using sub使用广播,切片,智能索引技巧,这些技巧用普通的C编写会很繁琐。我也欢迎关