PyTorch遇到的error1

RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #4 'mat1' TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor

>>阅读原文<<