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

Question

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

    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