Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

cast tensor type pytorch

y = y.to(torch.long)
# torch.long, torch.int16, torch.int32, torch.float16, etc.
Source by stackoverflow.com #
 
PREVIOUS NEXT
Tagged: #cast #tensor #type #pytorch
ADD COMMENT
Topic
Name
1+9 =