It seems that torch.tensor with device=torch.device('cuda') can't work with numpy arrays: ``` t = torch.tensor(np.array([5., 5., 5.]), device=torch.device('cuda')) t.device # returns 'cpu' ```