Оптимизация (Stiefel) многообразия с помощью TensorFlow
Я изо всех сил пытался максимизировать Tr(W'X'XW) с ограничением W'W = I в TensorFlow. Проблемной частью является ограничение, поскольку оно требует, чтобы веса оставались на многообразии Штифеля. Есть мягкий способ, который добавляет ограничение как лагранжево слагаемое к цели. Однако, это не точно, и реализация может быть грязной.
Мне известна библиотека Pymanopt, которая использует autodiff для оптимизации коллектора. Но мне все еще интересно, был ли принципиальный способ симулировать (Stiefel) оптимизацию коллектора в Tensorflow?