2

我正在寻找python中的内置函数。它应该计算 3D 数组的 frobenius 范数。我目前的做法是:

    np.sqrt(np.sum(np.square(x[:,:,:])))

但这对于我的数组大小来说太慢了。有任何想法吗?

提前致谢...

4

1 回答 1

2

np.linalg.norm将在高维数组上正常工作:

x = np.random.randn(100, 100, 100)
print np.allclose(np.linalg.norm(x), np.sqrt(np.sum(np.square(x))))
# True
于 2014-08-04T14:34:25.317 回答