torch.split()
1.7.0 compatibility fix
#7102
Merged
torch.split()
1.7.0 compatibility fix
#7102