How to convert a list or numpy array to a 1d torch tensor?

Question

I have a list (or, a numpy array) of float values. I want to create a 1d torch tensor that will contain all those values. I can create the torch tensor and run a loop to store the values.

But I want to know is there any way, I can create a torch tensor with initial values from a list or array? Also suggest me if there is any pythonic way to achieve this as I am working in pytorch.

Kashyap · Accepted Answer · 2017-03-20 15:11:59Z

31

These are general operations in pytorch and available in the documentation. PyTorch allows easy interfacing with numpy. There is a method called from_numpy and the documentation is available here

import numpy as np 
import torch 
array = np.arange(1, 11)
tensor = torch.from_numpy(array)

edited Mar 20, 2017 at 15:11

answered Mar 20, 2017 at 9:03

Kashyap

6,7492 gold badges25 silver badges21 bronze badges

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

Collectives™ on Stack Overflow

How to convert a list or numpy array to a 1d 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