Tensor 具象化举例

../../../_images/cube.svg

我们可以借助上面这张魔方( 图片来源 )来直观地理解 Tensor:

  • 首先,我们假设这个魔方是“实心同质”的,是一个存在于现实世界中的 Tensor;

  • 这个 Tensor 里面的每个元素的类型(dtype )都是一致的(方方正正的形状、外加一样的做工);

  • 而且这是一个维度(ndim )为 \(3\) 的结构,形状(shape )为 \((3, 3, 3)\) ;

  • 对应地,该 Tensor 的总元素个数(size )是 \(3*3*3=27\).

如果你将每种颜色代表一个值,而每个魔方块的值可以用其具有的颜色值之和来表示(此时中间块为零), 那么不同的魔方块就具有了各自的取值,就好像 Tensor 中的每个元素可以有自己的取值一样。 事实上,除了魔方以外,还有很多东西可以抽象成 Tensor 数据结构,意味着可以利用 MegEngine 进行相关的计算。