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