[PyTorch] GroupedTensor integration
#2600
Draft
Loading
GroupedTensor integration
#2600