我有个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
考虑一下这个二维数组: import numpy as np
a = np.array([[3,1,5,3],[1,4,8,2],[4,2,1,2],[9,2,4,4]]) 我知道如何找到每一行的最大条目这将返回array([5,8,4,9]) 然而,我想要做的是在某个索引n之后获得每一行的最大条目,即忽略第一个n条目,只在该行中每行的剩余条目中查找最大值。一个复杂的问题是:每一行的“切片”点是不同