pytorch

0a1b3be2 - chore: add unit test to verify split_by_tags output_type (#121262)

chore: add unit test to verify split_by_tags output_type (#121262) Add a test case as per https://github.com/pytorch/pytorch/pull/120361#issuecomment-1979163324 Pull Request resolved: https://github.com/pytorch/pytorch/pull/121262 Approved by: https://github.com/atalman