numpyマトリクスの和を求めるテクニック

573 ワード

python

多次元テンソル(マトリクス)のすべての要素の和を求めます.

import numpy as np
Y = np.random.randn(3,4,5)
sum_matrix = np.sum(np.reshape(Y,(Y.size,)))

テクニックはYを1次元ベクトルに変えることです.