Pytorch 保存tensor到.mat文件
import torch
import scipy.io as io
tensor = torch.ones(1, 2, 64)
tensor = tensor.cpu().detach().numpy()
#result = np.array(tensor)
io.savemat('save.mat', {'result':result})

本文介绍如何使用PyTorch将Tensor数据转换为MAT文件格式,并提供了完整的代码示例。通过将PyTorch的Tensor转换为NumPy数组,再利用SciPy库的savemat函数进行保存。
import torch
import scipy.io as io
tensor = torch.ones(1, 2, 64)
tensor = tensor.cpu().detach().numpy()
#result = np.array(tensor)
io.savemat('save.mat', {'result':result})

1729
1万+
3843
6343
1673

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