torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard
verified publisher icon pytorch

support dynamic dim for variable lengths in KJT

Open TroyGarden opened this issue 1 year ago • 1 comments

Summary:

context

  • previouls this helper function only supports len(kjt.lengths()) == batch_size * len(kjt.keys())
  • now it also supports that declaring len(kjt.lengths()) as a standalone dynamic dim

Differential Revision: D59194969

TroyGarden avatar Jun 30 '24 04:06 TroyGarden

This pull request was exported from Phabricator. Differential Revision: D59194969

facebook-github-bot avatar Jun 30 '24 04:06 facebook-github-bot