张量的维度

转发：https://blog.csdn.net/River_in_mind/article/details/105561569

张量的阶数有时候也称为维度，或者轴，轴这个词翻译自英文axis。

理解"沿着某个轴，提取某个轴的所有数据进行计算"：

import numpy as np
 
a = np.array([[1,2],[3,4]])
sum0 = np.sum(a, axis=0)
sum1 = np.sum(a, axis=1)
 
print(sum0)
print(sum1)

一个tensor的shape，从左到右依次为axis=0,axis=1,axis=2...

axis=0时，np.sum(a,axis=0),将0阶的数据全部加起来，即每列所有行加起来

axis=1时，np.sum(a,axis=1),将1阶的数据全部加起来，即每行所有列加起来