python - 如何将 torch 张量转换为 Pandas 数据帧？

Question

我想将火炬张量转换为 Pandas 数据帧，但使用 pd.DataFrame我得到了一个充满张量而不是数值的数据框。

import torch
import pandas as  pd
x = torch.rand(4,4)
px = pd.DataFrame(x)

这是我点击 px 时得到的结果在变量资源管理器中:

0   1   2   3
tensor(0.3880)  tensor(0.4598)  tensor(0.4239)  tensor(0.7376)
tensor(0.4174)  tensor(0.9581)  tensor(0.0987)  tensor(0.6359)
tensor(0.6199)  tensor(0.8235)  tensor(0.9947)  tensor(0.9679)
tensor(0.7164)  tensor(0.9270)  tensor(0.7853)  tensor(0.6921)

Answer 1

我通过首先将 torch 转换为 numpy 找到了一种可能的方法:

import torch
import pandas as  pd

x = torch.rand(4,4)
px = pd.DataFrame(x.numpy())