1. 程式人生 > >numpy 矩陣求和小技巧

numpy 矩陣求和小技巧

求多維張量(矩陣)所有元素之和。

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

技巧就是把Y變為一維向量。