Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import torch
- def my_main_function(device):
- t = torch.tensor(22)
- t = t.to(device)
- print(t)
- print(t.device)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement