Extract tensor from string

Question

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.

End genocide - save Gaza · Accepted Answer · 2021-03-23 15:52:01Z

3

You can use eval:

import torch.tensor as tensor

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

edited Mar 23, 2021 at 15:52

End genocide - save Gaza

25k10 gold badges114 silver badges133 bronze badges

answered Jul 9, 2020 at 12:26

Dishin Goyani

7,7533 gold badges33 silver badges42 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Belkacem Thiziri Over a year ago

Does the eval function recognize the tensor inside the string? this is awesome!

Dishin Goyani Over a year ago

eval will run string as python code. and we have added alias for tensor before (import torch.tensor as tensor) so it worked.

Collectives™ on Stack Overflow

Extract tensor from string

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related