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 中运行得非常好。但对于旧版本，您可以使用 .tolist() 方法将 numpy 数组转换为列表，如下所示，以消除错误。

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