pytorch的save函数，保存了整个tensor

最新推荐文章于 2025-10-15 14:23:31 发布

翻译最新推荐文章于 2025-10-15 14:23:31 发布 · 353 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://pytorch.org/docs/stable/notes/serialization.html#saving-and-loading-tensors-preserves-views

文章标签：

#pytorch #人工智能 #python

探讨如何在PyTorch中正确保存部分Tensor，避免覆盖全量数据，需用`clone()`方法。

部署运行你感兴趣的模型镜像

1、当使用如下代码，想保存一个tensor的一部分时


large = torch.arange(1, 1000)
small = large[0:5]
torch.save(small, 'small.pt')
loaded_small = torch.load('small.pt')
loaded_small.storage().size()
# 999

最后保存的结果却不是，large[0:5]，而是整个large,
想要解决这个问题，需要加一个使用tensor的clone函数。

small.clone()

完整保存代码

large = torch.arange(1, 1000)
small = large[0:5]
torch.save(small.clone(), 'small.pt')  # saves a clone of small
loaded_small = torch.load('small.pt')
loaded_small.storage().size()
# 5