且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

从字符串中提取张量

更新时间:2022-10-15 09:45:55

您可以使用 eval:

import torch.tensor 作为张量评估(张量列表)>>>张量([-1.6975, 0.0176, -2.4441, -2.3994, -0.6207])

Is it possible to extract directly the tensor included in this string tensor([-1.6975e+00, 1.7556e-02, -2.4441e+00, -2.3994e+00, -6.2069e-01])? I'm looking for some tensorflow or pytorch function that can do it, like the ast.literal_eval function does for dictionaries and lists.

If not, could you provide a pythonic method, please?

I'm thinking about something like this:

tensor_list = "tensor([-1.6975e+00,  1.7556e-02, -2.4441e+00, -2.3994e+00, -6.2069e-01])"
str_list = tensor_list.replace("tensor(", "").replace(")", "")
l = ast.literal_eval(str_list)
torch.from_numpy(np.array(l))

But I'm not sure this is the best way.

You can use eval:

import torch.tensor as tensor

eval(tensor_list)
>>> tensor([-1.6975,  0.0176, -2.4441, -2.3994, -0.6207])