Train a diagonal unitary matrix?

I leave here the reference to the other conversation :slight_smile: