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

The simplest and cleanest method I know:

IntTensor.item()

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

• PyTorch Docs