Solved by albanD in post #2 
              
              
                That is expected, all tensors are of type torch.Tensor. 
You can use .type() to get a more precise type. And attributes like .is_cuda.