Самый быстрый способ рассчитать сумму строки "v.t @ v" матрицы ⇐ Python

[Anonymous]

У меня есть матрица 1000 x 500. Для каждой строки я хочу рассчитать точечный продукт между транспонированием вектора строки и самого вектора строки (который должен быть скалярным) и суммировать по всем рядам. Прямо сейчас я использую < /p>

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

import numpy as np

A = np.random.random((1000, 500))
res = sum(A[i].T @ A[i] for i in range(A.shape[0]))

Поскольку это узкое место в моем алгоритме, я удивляюсь, есть ли более быстрые способы сделать это, омно -омпейское решение.

Подробнее здесь: https://stackoverflow.com/questions/769 ... f-a-matrix