2019年9月19日
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 阅读全文
posted @ 2019-09-19 22:40 The-Chosen-One 阅读(20067) 评论(0) 推荐(3) 编辑