Pytorch - проецирует каждую строку тензора в пространство столбцов другого тензора

Question

В настоящее время у меня есть тензор A и тензор U , где U является ортогональной матрицей и имеет полный ранг (так что ее столбцы набор базисов пространства столбцов U , и все столбцы, скажем, u_i , имеют норму 1).

Я пытаюсь вычислить проекция каждой строки A на пространство столбца U , используя формулу из этого поста .

То есть для вычисления Proj (А). Есть ли удобные функции или более эффективные операции для достижения этой цели? Спасибо.

Answer 1

Если у вас уже есть единичная норма в каждом столбце матрицы проекции, достаточно просто

torch.mm(A,U)

.