张量是一棵树 长久以来,张量和其中维度的概念把我搞的晕头转向. 一维的张量是数组,二维的张量是矩阵,这也很有道理. 但是给一个二维张量,让我算出它每一行的和,应该用 sum(dim=0) 还是 sum(dim=1)? 这个问题还得让我想个一会儿. 更别说四维的张量是什么,一百维的张量又是什么,这种问题了,我不知道,想想就头大. 但是直到把张量看成一棵树,许多问题就迎刃而解~ 如下图所示,分别表示三种不同形状的张量: 基本规律是: 不算最上边的树根节点,剩下的节点有几层,那这个张量就是几维的.(换…