unable to get repr for class 'torch.tensor'