3️⃣ Pytorch 기초_boolean 인덱싱

 

1) Boolean 인덱싱

 

torch에서 boolean 인덱싱을 활용하면 나중에 transformer 마스킹 등을 할 때 매우 유용하게 쓰일 수 있다고 한다. 

boolean을 사용하면 True 인 성분만을 가져와서 새로운 행렬을 만들 수도 있다.

 

 

• A[B,:]: B의 각 요소에 대응하는 위치에 있는 A의 행을 선택하여 새로운 텐서를 생성한다. B에서 True 에 해당하는 위치에 있는 A의 행들을 선택해서 새로운 텐서를 만든다.

A=torch.tensor([[1,2],[3,4],[5,6],[7,8]])
print(A[ torch.tensor([[False,True],[False,False],[False,False],[False,False]]) ]) 
print(A[A==2])
print(A[ [True,False,False,False], [False,True] ])

• line 2: True 에 해당하는 2 를 반환
• line 3: A 텐서에서 조건을 만족하는 원소 (값이 2인 원소)를 선택하고 그걸로 새로운 텐서를 반환
• line 4: True 인 A의 1행의 2열 선택하여 반환

tensor([2])
tensor([2])
tensor([2])