cast tensor type pytorch

Solutions on MaxInterview for cast tensor type pytorch by the best coders in the world

showing results for - "cast tensor type pytorch"
Elliot
17 Mar 2016
1y = y.to(torch.long)
2# torch.long, torch.int16, torch.int32, torch.float16, etc.