Picassooo

摘要： 把数据从CPU迁移到GPU时，可以用.cuda()方法，也可以用.to(device)方法。示例如下。 .cuda()方法 import torch import time t = time.time() b = torch.rand([1024, 1024, 10]) b = b.cuda() p 阅读全文

摘要： 参考一 浅谈 PyTorch 中的 tensor 及使用 该博文分为以下6个部分： tensor.requires_grad torch.no_grad() 反向传播及网络的更新 tensor.detach() CPU and GPU tensor.item() torch.detach()和tor 阅读全文