How to cast a 1-d IntTensor to int in Pytorch

问题

I get a 1-D IntTensor,but i want to convert it to a integer. I try it by this method:

print(dictionary[IntTensor.int()])

but got an error:

KeyError: Variable containing:
 423
[torch.IntTensor of size 1]

Thanks~

回答1:

You can use:

print(dictionary[IntTensor.data[0]])

The key you're using is an object of type autograd.Variable. .data gives the tensor and the index 0 can be used to access the element.

回答2:

The simplest and cleanest method I know:

IntTensor.item()

From PyTorch docs:

"Returns the value of this tensor as a standard Python number. This only works for tensors with one element. For other cases, see :meth:~Tensor.tolist"

回答3:

torch.tensor('variable',dtype=torch.int8)

try this

来源：https://stackoverflow.com/questions/47588682/how-to-cast-a-1-d-inttensor-to-int-in-pytorch