numpy 矩阵求和小技巧

来源:互联网 发布:centos配置rsyslog 编辑:程序博客网 时间:2024/06/05 00:37

求多维张量(矩阵)所有元素之和。

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

技巧就是把Y变为一维向量。

原创粉丝点击