json 如何将填充复数的torchTensor序列化？

xfyts7mz 于 2023-02-26 发布在其他

关注(0)|答案(1)|浏览(101)

我有这样一些数据：

tensor([[0.9938+0.j, 0.1109+0.j],
        [1.0000+0.j, 0.0000+0.j],
        [0.9450+0.j, 0.3272+0.j],
        [0.9253+0.j, 0.3792+0.j],
        [0.9450+0.j, 0.3272+0.j],
        [0.9028+0.j, 0.4300+0.j],
        [0.8776+0.j, 0.4794+0.j],
        [0.9253+0.j, 0.3792+0.j],
        [0.9253+0.j, 0.3792+0.j],
        [0.9985+0.j, 0.0555+0.j]])

我想把这些数据保存到JSON文件中。
我试着通过data.numpy().tostring()将其转换为字符串，但它给了我一个错误，说TypeError: Object of type bytes is not JSON serializable。
有没有办法把它写入JSON，然后再读回来？

JSON

来源：https://stackoverflow.com/questions/75517311/how-can-i-serialize-torch-tensor-filled-with-complex-number

1条答案

按热度按时间

juzqafwq1#

你好pytorch可以保存字典使用保存，你可以加载与.load
下面是一个例子：

import torch

save_dict = {
   "test": torch.randn(1, 100),
   "test2": torch.zeros(100),
   "test3": torch.ones(100)
}

torch.save(save_dict, 'test.pt')
torch.load('test.pt')

赞(0）回复(0）举报 2023-02-26

我来回答

json 如何将填充复数的torchTensor序列化？

1条答案

相关问题

热门标签

最新问答