标签： optimization tensorflow manifold
我正努力在TensorFlow中使用约束W＆＃39; W = I来最大化Tr（W＆＃39; X＆＃39; XW）。有问题的部分是约束，因为它需要权重保持在Stiefel流形上。有一种软方式可以将约束作为拉格朗日项添加到目标中。但是，这并不准确，实施可能会很混乱。
我知道Pymanopt库使用autodiff进行多种优化。但我仍然想知道是否有一种在Tensorflow中模拟（Stiefel）流形优化的主要方法？