張量，或tensor，可以看作是向量、矩陣的自然推廣，我們用張量來表示廣泛的資料型別。

張量的階數有時候也稱為維度，或者軸，軸這個詞翻譯自英文axis。

譬如一個矩陣[[1,2],[3,4]]，是一個2階張量，有兩個維度或軸。

沿著第0個軸你看到的是[1,2]，[3,4]兩個向量，相當於數學中一個2x2的矩陣，按行取出，每一行作為一個向量；

沿著第1個軸你看到的是[1,3]，[2,4]兩個向量，相當於數學中一個2x2的矩陣，按列取出，每一列作為一個向量；

import numpy as np

a = np.array([[1,2],[3,4]])
s0 = np.sum(a, axis=0)  #即[1,2]，[3,4]
 
相加
s1 = np.sum(a, axis=1)  #即[1,3]，[2,4]相加

print(s0)
print(s1)

>>> [4, 6]

>>> [3, 7]