求多维张量(矩阵)所有元素之和。
import numpy as np
Y = np.random.randn(3,4,5)
sum_matrix = np.sum(np.reshape(Y,(Y.size,)))
技巧就是把Y变为一维向量。
求多维张量(矩阵)所有元素之和。
import numpy as np
Y = np.random.randn(3,4,5)
sum_matrix = np.sum(np.reshape(Y,(Y.size,)))
技巧就是把Y变为一维向量。