python - 寻找 numpy.linalg.norm 的 3D 版本

Question

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

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

但这对于我的数组大小来说太慢了。有什么想法吗？

提前致谢...

Answer 1

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