摘要：
tensor.equal()方法： # 该方法用于比较两个tensor是否一样，一样则返回True否则为False a = torch.tensor([1,2,3,4]) b = torch.tensor([1,2,3,4]) print(a.equal(b)) # 返回True tensor.eq 阅读全文
摘要：
数据类型之间的转换 numpy转list list = ndarray.tolist() tensor转numpy ndarray = tensor.numpy() # 注意gpu上的tensor不能直接转为numpy，应先放回cpu中 ndarray = tensor.cpu().numpy() 阅读全文
摘要：
与GPU的相关操作 查看gpu是否可用 torch.cuda.is_available() # 可用返回True，不可用返回False 返回可用的CPU的数量 torch.cuda.device_count() 返回gpu的名字，设备索引默认从0开始 torch.cuda.get_device_na 阅读全文
摘要：
在pytorch中有两种模型保存方式，如下所示： # 第一种：只保留模型参数： torch.save(model.state_dict(),path) # 保存 model.load_state_dict(torch.load(path)) # 加载（需提前定义模型） # 第二种：保存整个模型： t 阅读全文