Pytorch에서 int로 1-d IntTensor를 캐스팅하는 방법

Question

1-D IntTensor를 얻지 만 정수로 변환하고 싶습니다. 나는이 방법으로 그것을 시도 :

print(dictionary[IntTensor.int()]) 

하지만 오류가있어 :

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

감사합니다 ~를

Answer 1

당신은 사용할 수 있습니다

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

의 핵심을 ' 다시 사용하는 것은 autograd.Variable 유형의 객체입니다. .data은 텐서와 인덱스를 제공합니다. 0은 요소에 액세스하는 데 사용할 수 있습니다.