weixin_42080885的博客

python、numpy 、pytorch高级切片索引，[]放数组。 只实验了二维数组，表示对同一个维度进行多次切片，二维数组每行的值为每次的切片索引，有n行就切片n次，同时把该维度切成n个维度。 numpy和pytorch结果一样，只放了numpy代码。 直接上结果： import numpy as np a=np.random.randn(2,3,4) b=[[1,2],[0,1]] c=a[:,b,:] print(a) print("c--------") print(c) 结果： [[[-1.