I have tested torch==1.7.0, torch==1.9.0, and torch==1.10.0
I have the following error. Can I get away from this?

torch.tensor([torch.tensor(0.), torch.tensor(1.)]) # works
torch.tensor([torch.tensor(0), torch.tensor(1)], dtype=torch.int32) # works
torch.tensor([torch.tensor(0.), torch.tensor(1.)], dtype=torch.int32) # fails
> TypeError: only integer tensors of a single element can be converted to an index