[Python] IndexError: tensors used as indices must be long, byte or bool tensors 문제 해결

subgraph를 추출하는 과정에서 만난 에러!

그래프 내 특정 타겟 노드에 대해서만 다음과 같은 에러가 발생하는 것을 확인할 수 있었다.

 

IndexError: tensors used as indices must be long, byte or bool tensors

 

tensor를 인덱스로 사용 시 반드시 long, byte, bool이어야 한다는 것.

코드를 확인해보니 특정한 경우에는 인덱스로 사용되는 tensor가 empty하여 이런 에러가 발생한 것을 확인할 수 있었다.

그렇다면 이런 경우는 따로 예외처리를 해야할까?

 

물론 예외처리를 할 수도 있지만 인덱스로 사용되는 tensor를 numpy 객체로 변환하여 사용하여 문제없이 사용가능하다!