3.1 Tensor的定义
PyTorch最基本的操作对象是张量,张量的英文是Tensor,表示一个多维的矩阵。张量的维数常被描述为阶,如一阶张量、二阶张量。零阶张量就是一个标量,一阶张量就是一个矢量,二阶张量就是一般的矩阵(二维),多阶张量就相当于多维的矩阵数组。图3.1形象地描述了零阶—三阶张量。
![](https://epubservercos.yuewen.com/1CD67A/18002371308017606/epubprivate/OEBPS/Images/39396_45_1.jpg?sign=1739545849-iiRG9VuB6XA866Okp5v48k8NHLUjFyXZ-0-631e2367c68d71d4f07e8c1e98c91554)
图3.1 零阶—三阶张量
张量的三个基本属性如下。
(1)阶(rank):维数。
(2)形状(shape):行和列的数目。
(3)类型(type):元素的数据类型。
现在将三阶张量用一个正方体来表示,如图3.2所示。
![](https://epubservercos.yuewen.com/1CD67A/18002371308017606/epubprivate/OEBPS/Images/39396_46_1.jpg?sign=1739545849-o0xWqWsw2kLeabyr5YANNzqnq8o58peJ-0-95b1d58084640feb63458243d3e8c2ba)
图3.2 三阶张量
这样可以进一步生成更高阶的张量,四阶张量、五阶张量如图3.3所示。
![](https://epubservercos.yuewen.com/1CD67A/18002371308017606/epubprivate/OEBPS/Images/39396_46_2.jpg?sign=1739545849-FCzKSP6fh1bk6rv0XNfXQ3Uoidsb86ik-0-12d77af7494eefd6b171f814e0ce7939)
图3.3 四阶张量、五阶张量
张量常用的数据类型如表3.1所示,包括整型和浮点型。
表3.1 张量常用的数据类型
![](https://epubservercos.yuewen.com/1CD67A/18002371308017606/epubprivate/OEBPS/Images/39396_46_3.jpg?sign=1739545849-JABHabuLDJUx94mgryCnu0IMluPlaZLb-0-f64009a09d7d04a434af92002324d0a1)