jaxopt icon indicating copy to clipboard operation
jaxopt copied to clipboard

Projected gradient for multidimensional array

Open rkruegs123 opened this issue 1 year ago • 0 comments

Hi -- I have an array that is mxnxn. Each of the m nxn matrices represents a single "frame." I wish to enforce that each nxn matrix/frame is a doubly-stochastic matrix. For a single nxn matrix, I could use projected gradients with the birkhoff projection. However, I would like to apply this constraint to every nxn matrix.

Does anyone know how to best implement this?

rkruegs123 avatar May 14 '24 20:05 rkruegs123