Самый быстрый способ вычисления суммы квадратных элементов результата умножения матрицы? ⇐ Python

[Anonymous]

Я хочу сделать что -то подобное в Python < /p>
sum(square(matmul(A, B)))
< /code>
Есть различные способы достижения такого поведения, например, < /p>
sum(np.square(np.matmul(A, B)))
< /code>
или < /p>
np.lingalg.norm(np.matmul(A, B)) ** 2
< /code>
Я еще не профилировал его, но я подозреваю, что второй вариант будет быстрее (меньше OPS, Fused Operations и т. Д.). < /p>
Есть ли другие подходы, которые потенциально могут победить это? Я хотел бы профилировать различные подходы к решению этого. Я не ограничен Numpy.

Подробнее здесь: https://stackoverflow.com/questions/790 ... -matrix-mu