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）的數據，示例如下圖：

>>阅读原文<<