Reformer Data Rotation - Use unique vectors for random rotations.
Reformer Data Rotation - Use unique vectors for random rotations.

If the number of vectors for rotation exceeds the total number of vectors
available, then sample with replacements.

PiperOrigin-RevId: 277374631

google/trax,google/trax