#範例7-1：下面的程式碼可以幫助理解張量tensor“沿着某個軸”是什麼意思
import numpy as np
a = np.array([[1,2],[3,4]]) 
sum0 = np.sum(a, axis=0) 
sum1 = np.sum(a, axis=1)
print('矩陣a=\n', a)
#2 階張量，有兩個維度或軸，沿著第 0 個軸= [1,2]、[3,4] 兩個向量
print('2 階張量，有兩個維度或軸，沿著第0個軸==', a[0,:],a[1,:])
print('2 階張量，有兩個維度或軸，沿著第1個軸==', a[:,0],a[:,1])
#2 階張量，有兩個維度或軸，沿著第 0 個軸= [1,2]、[3,4] 兩個向量
print('2 階張量，有兩個維度或軸，沿著第 0 個軸，總和=', sum0) 
#2 階張量，有兩個維度或軸，沿著第 1 個軸= [1,3]、[2,4] 兩個向量
print('2 階張量，有兩個維度或軸，沿著第 1 個軸，總和=', sum1)