



在编程实践中,张量可以被视为多维数组。例如,标量是零维张量,向量是一维张量,矩阵是二维张量,RGB图像则是三维张量(颜色通道×高度×宽度)。因此,对64幅图像的批处理构成了一个四维张量(64×颜色通道×高度×宽度)。
代码清单2.2展示了表示矩阵的PyTorch代码。
代码清单2.2 通过PyTorch来表示矩阵
代码清单2.3展示了如何通过PyTorch实现矩阵的切片和分块。
代码清单2.3 矩阵的切片和分块
代码清单2.4 展示了如何用PyTorch表示张量和图像。
代码清单2.4 用PyTorch表示张量和图像