1。计算矩阵的size和dimension
两个函数: array.ndim
和 np.size(array, axis)
import numpy as np
# 定义一个二维张量
X=np.array([[1,2,3,4],
[5,6,7,8],
[9,10,11,12]])
dim=X.ndim # 计算X的维度 = 2
number=X.size # 计算 X 中所有元素的个数 = 12
X_row=np.size(X,0) #计算 X 一行元素的个数 = 4
X_col=np.size(X,1) #计算 X 一列元素的个数 = 3
print("number:",number)
print("X_row:",X_row)
print("X_col:",X_col)
<<
number: 12
X_row: 3
X_col: 4