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

[Anonymous]

Я хочу сделать что-то подобное на Python

Код: Выделить всё

sum(square(matmul(A, B)))

Существует несколько способов добиться такого поведения, например:

Код: Выделить всё

sum(np.square(np.matmul(A, B)))

или

Код: Выделить всё

np.lingalg.norm(np.matmul(A, B)) ** 2

Я еще не профилировал его, но подозреваю, что второй вариант будет быстрее (меньше операций, объединенных операций и т. д.).
Есть ли другие подходы, которые потенциально могли бы победить это? Я хотел бы описать различные подходы к решению этой проблемы. Я не ограничен numpy.

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