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:,:,:]：取第一个维度的前两个数据，取第

>>阅读原文<<