How to convert numpy array into LongTensor type Variable in PyTorch?

Question

I am trying to convert numpy array into PyTorch LongTensor type Variable as follows:

import numpy as np
import torch as th

y = np.array([1., 1., 1.1478225, 1.1478225, 0.8521775, 0.8521775, 0.4434675])
yth = Variable(th.from_numpy(y)).type(torch.LongTensor)

However the result I am getting is a rounded off version:

tensor([ 1,  1,  1,  1,  0,  0,  0])

How can I keep the precision of numpy array while getting LongTensor variable?

Expected result should be:

tensor([1., 1., 1.1478225, 1.1478225, 0.8521775, 0.8521775, 0.4434675])

benjaminplanche · Accepted Answer · 2018-06-19 12:21:19Z

1

LongTensor represents tensors with values of type long / int64 (c.f. table in doc). Your float values are thus converted (i.e. rounded) to integers.

To keep float values, use FloatTensor (float32) or DoubleTensor (float64) instead.

answered Jun 19, 2018 at 12:21

benjaminplanche

15.2k5 gold badges64 silver badges73 bronze badges

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

Collectives™ on Stack Overflow

How to convert numpy array into LongTensor type Variable in PyTorch?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related