Pytorch Tensor帮助LongTensor

Question

a=[1,2,3];
context_var = autograd.Variable(torch.LongTensor(a))

这是一个错误

RuntimeError: tried to construct a tensor from a int sequence, but found an item of type numpy.int32 at index

我无法弄清楚如何克服这个问题。

Answer 1

您的代码在最新版本的pytorch中运行得非常好。但对于旧版本，您可以使用numpy方法将.tolist()数组转换为列表，如下所示，以消除错误。

a=[1,2,3];
context_var = autograd.Variable(torch.LongTensor(a.tolist())) 

Answer 2

对我来说很好：

a=[1,2,3]
print(torch.autograd.Variable(torch.LongTensor(a)))
b = np.array(a)
print(torch.autograd.Variable(torch.LongTensor(b)))

输出：

Variable containing:
 1
 2
 3
[torch.LongTensor of size 3]

Variable containing:
 1
 2
 3
[torch.LongTensor of size 3]

我使用的是Python 3.6.2，火炬0.2.0.post3和numpy 1.13.3。