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

后端 未结 3 1214
独厮守ぢ
独厮守ぢ 2021-02-13 11:46

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:

3条回答
  •  灰色年华
    2021-02-13 12:17

    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.

提交回复
热议问题