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

torch pytorch
Wasi Ahmad picture Wasi Ahmad · Mar 20, 2017 · Viewed 20.4k times · Source

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.

Answer

Kashyap picture Kashyap · Mar 20, 2017

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)

Related questions

How does the "view" method work in PyTorch?

I am confused about the method view() in the following code snippet. class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2,2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = …

Read More
Torch sum a tensor along an axis

ipdb> outputs.size() torch.Size([10, 100]) ipdb> print sum(outputs,0).size(),sum(outputs,1).size(),sum(outputs,2).size() (100L,) (100L,) (100L,) How do I sum over the columns instead?

Read More
Using CUDA with pytorch?

I have searched on here but I found only outdated posts. I want to run the training on my GPU. I found on some forums that I need to apply .cuda() on anything I want to use CUDA with (I've …

Read More