How to convert numpy array(float data) to torch tensor?

Question

test = ['0.01171875', '0.01757812', '0.02929688']  
test = np.array(test).astype(float)   
print(test)  
->[0.01171875 0.01757812 0.02929688]


test_torch = torch.from_numpy(test)  
test_torch  
->tensor([0.0117, 0.0176, 0.0293], dtype=torch.float64)

It looks like from_numpy() loses some precision there... If I want to convert this float data exactly the same, what kind of functions do I use?

Does this answer your question? How to convert a list or numpy array to a 1d torch tensor? — Dwa
– Dwa, Commented Aug 4, 2021 at 15:27
@Dwa I don't think this is the issue here, OP already knows how to convert from NumPy to PyTorch. However, OP is concerned with - what seems to be - a lack of precision after conversion. — Ivan
– Ivan, Commented Aug 4, 2021 at 18:38

Ivan · Accepted Answer · 2021-08-04 15:21:20Z

5

The data precision is the same, it's just that the format used by PyTorch to print the values is different, it will round the floats down:

>>> test_torch = torch.from_numpy(test)
>>> test_torch
tensor([0.0117, 0.0176, 0.0293], dtype=torch.float64)

You can check that it matches your original input by converting to a list with tolist:

>>> test_torch.tolist()
[0.01171875, 0.01757812, 0.02929688]

answered Aug 4, 2021 at 15:21

Ivan

41.3k9 gold badges78 silver badges120 bronze badges

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

Collectives™ on Stack Overflow

How to convert numpy array(float data) to torch tensor?

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related