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

Question

目前，我有一个张量 A 和一个张量 U ，其中 U 是正交矩阵，并且具有完整秩（因此它的列是 U 列空间的基础，并且所有列（例如 u_i ）的范数均为1）。

我正在尝试使用this post中的公式来计算 A 的每一行到 U 的列空间上的投影。

计算Proj（A）。 是否有任何便捷的功能或更好的操作来实现这一目标？ 谢谢。

Answer 1

如果在投影矩阵的每一列中已经有单位范数，只需简单地

2

应该足够。