325-张量

在 pytorch 中会用到张量,这个概念来源于数学:

数学上:张量是标量、向量、矩阵,以及更高维度的延伸,描述多维数据的量(N维数据的统称);

  • 0 阶张量(0 维):标量,单个数字,如 53.14

  • 1 阶张量(1 维):向量 / 一维数组,一串数字,如 [1,2,3]

  • 2 阶张量(2 维):矩阵,行列二维表格,如 [[1,2],[3,4]]

  • 3 阶张量(3 维):立方体数据,例如一张 RGB 图片 [高度,宽度,通道]

  • N 阶张量(N 维):更高维度,比如一批图片 [批次,高,宽,通道]