Cast a Tensor to another Type in PyTorch. ... import torch # Create a float tensor y = torch.randn(5, 5) # Cast the flost tensor to int ...
確定! 回上一頁