토치의 longtensor 형식의 인덱스를 이진 선택 마스크로 변환

Question

다른 텐서에서 원하는 모든 인덱스가 포함 된 LongTensor가 있습니다. 이 LongTensor를 어떻게 선택 마스크로 사용할 수있는 ByteTensor로 변환 할 수 있습니까?

가정,

th> imageLabels:size() 
17549 
    3 
[torch.LongStorage of size 2] 

                     [0.0001s] 
th> indices 
    1 
22 
32 
[torch.LongTensor of size 3] 

는 내가 현재 위치에서 imageLabels에서 일부 값을 변경할 수 있도록 [index] 표기법을 사용하여 imageLabels에 액세스 할 수있는 방법이 필요합니다.

이렇게 할 방법이 있습니까? 필자가 문서에서 이해하는 한 : index : 좁은 작업은 완전히 새로운 Tensor를 반환합니다.

Answer 1

나는 결국 indexFill을 사용했다.

targetTensor :

• 첫 번째 인수는 치수가 indexFill (0 1, 인덱스),
• 인덱스 우리
• 0 관련 모든 인덱스를 포함 LongTensor되는 값인 가득 따르다. 어떤 숫자도 가능합니다.

희망이 있습니다. 그것의 모든 워드 프로세서에. 우리는 참을성있게 읽어야합니다.

Answer 2

정확도, : 색인, 좁은 거리의 새 텐서를 반환합니다. 새 텐서는 here과 같은 원본 저장소를 사용합니다. "좁은 방법의 경우 반환 된 텐서를 선택하고 원본과 동일한 저장소를 공유합니다."