Answers for "change tensor type pytorch"

0

change tensor type pytorch

y = y.to(torch.long)
# torch.long, torch.int16, torch.int32, torch.float16, etc.
Posted by: Guest on May-29-2021

Python Answers by Framework

Browse Popular Code Answers by Language