import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
tensor = torch.randn(3, 3).to(device)  # Moves the tensor to GPU


print(tensor)
print(device)