Pytorch-将一个张量的每一行投影到另一个张量的列空间

目前，我有一个张量a和一个张量U，其中U是一个正交矩阵，并且是满秩的（因此它的列是U的列空间的一组基，并且所有列，比如U_I，都有一个范数1）

我试图计算A的每一行在U的列空间上的投影，使用的公式是

也就是说，计算Proj（A）。 是否有任何方便的功能或更好的操作来实现这一点？

---

谢谢

如果在投影矩阵的每列中都有一个单位范数，只需

torch.mm(A,U)

应该足够了