购买
下载掌阅APP,畅读海量书库
立即打开
畅读海量书库
扫码下载掌阅APP

2.4 Python代码:如何用PyTorch表示矩阵、张量和图像

在编程实践中,张量可以被视为多维数组。例如,标量是零维张量,向量是一维张量,矩阵是二维张量,RGB图像则是三维张量(颜色通道×高度×宽度)。因此,对64幅图像的批处理构成了一个四维张量(64×颜色通道×高度×宽度)。

代码清单2.2展示了表示矩阵的PyTorch代码。

代码清单2.2 通过PyTorch来表示矩阵

代码清单2.3展示了如何通过PyTorch实现矩阵的切片和分块。

代码清单2.3 矩阵的切片和分块

代码清单2.4 展示了如何用PyTorch表示张量和图像。

代码清单2.4 用PyTorch表示张量和图像 yUrfr9P6ap/+GLdA3FJ5vOTh0H06H0duJIr/SuIEglYHV2lvtV0mDfSEdeMZYMXO

点击中间区域
呼出菜单
上一章
目录
下一章
×