PyTorch入门实战教程笔记(五):基础张量操作2

PyTorch入门实战教程笔记(五):基础张量操作2 包括:索引与切片和 维度变换 索引和切片: 使用函数torch.rand()来创建一个数据,比如a = torch.rand(4,3,28,28),即为Batch size为4(即4张图片)的28×28的RGB图像,这也是CNN中最常用的,那么a[0]指的就是索引的第一张图片,a[0,1]指的是第一张图片第一个通道(如:R)的数据,示例如下图:
相关文章
相关标签/搜索