torchrec
torchrec copied to clipboard
Add option to pass inverse indices to new KJT after permute
Summary: allow inverse indices to be included in new KJT after permute. For modifying vbe KJT outside of input dist.
Differential Revision: D53732141
This pull request was exported from Phabricator. Differential Revision: D53732141