tensor怎么保存为npy文件
时间: 2023-07-28 15:07:45 浏览: 283
将tensorflow的ckpt模型存储为npy的实例
可以使用numpy库中的`save()`函数将tensor保存为npy文件。具体操作如下:
```python
import numpy as np
import torch
# 创建一个tensor
tensor = torch.randn(2, 3)
# 将tensor转换为numpy数组
array = tensor.numpy()
# 将numpy数组保存为npy文件
np.save('tensor.npy', array)
```
这样就可以将tensor保存为npy文件了。在需要使用时,可以使用numpy库中的`load()`函数将npy文件加载为numpy数组,再将其转换为tensor。
阅读全文