Torch.matmul(S, v), где S — симметрично, а v — вектор: как ускорить вычисления? ⇐ Python

[Anonymous]

Пусть S — симметричная матрица nxn, а v — одномерный вектор.
Нам нужно эффективно вычислить вектор (S x v) внутри функции потерь pytorch.
Знаете ли вы, есть ли способ сохранить градуировки тензоров и использовать быструю и специализированную функцию torch.matmul() для симметричных входных матриц?
Извините за мое незнания, но нам сказали использовать только методы pytorch внутри реализации потерь.

Подробнее здесь: https://stackoverflow.com/questions/798 ... d-up-compu