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

Question

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~

Answer 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.

Answer 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"

Answer 3

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

try this