将火炬张量转换为字典列表

问题描述 投票:0回答:0

我有一些这样的数据:

tensor([[ 0.2054+0.7081j,  0.6586+0.1506j],
        [ 0.4596+0.7250j, -0.1743+0.4825j]])

我想把它转换成这样的东西:

[[{"real":0.2054,"img":0.7081},{"real":0.6586,"img":0.1506}],
 [{"real":0.4596,"img":0.7250},{"real":-0.1743,"img":0.4825}]]

其中每个值都替换为表示其实部和虚部的字典,以便我可以将其保存到 JSON 文件中。我应该如何转换它并在以后“取消转换”它?

我想一些类似的代码可能会完成这项工作,但我不熟悉 Python 和 torch 的语法。

python json torch
© www.soinside.com 2019 - 2024. All rights reserved.