pytorch張量數據索引切片與維度變換操作大全

(1-1)pytorch張量數據的索引與切片操作 1、對於張量數據的索引操作主要有以下幾種方式： a=torch.rand(4,3,28,28):DIM=4的張量數據a (1)a[:2]:取第一個維度的前2個維度數據(不包括2)； (2)a[:2,:1,:,:]：取第一個維度的前兩個數據，取第2個維度的前1個數據，後兩個維度全都取到； (3)a[:2,1:,:,:]：取第一個維度的前兩個數據，取第

>>阅读原文<<