保存加载数据
torch.save(tensor, path)
tensor = torch.load(path, map_location="cpu")
比较tensor
torch.equal(a, b) #完全相等
torch.allclose(a, b, atol=5e-4, rtol=5e-4) #允许误差在5e-4以内
torch.save(tensor, path)
tensor = torch.load(path, map_location="cpu")
torch.equal(a, b) #完全相等
torch.allclose(a, b, atol=5e-4, rtol=5e-4) #允许误差在5e-4以内
1516
303

被折叠的 条评论
为什么被折叠?
