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