pytorch boolean indexing

pytorch 만의 특별한 기능

A=torch.tensor([[1,2,3,4],[5,6,7,8]])
print(A==5) # 리스트와 달리 각 성분에 대해 비교해줌
print(A[A==5]) # True, False가 담긴 행렬로 인덱싱 가능

A[A==5] = 777	# 5를 777로 변경
print(A) 

A=torch.tensor([[1,2],[3,4],[5,6],[7,8]])
B=torch.tensor([True, False, False, True]) # 참고로 그냥 리스트여도 됨
print(A[B,:]) # 0행, 3행 슬라이싱

b=torch.tensor([1,2,3,4])
print(b[ [True,True,False,False] ])
c=[1,2,3,4]
# c[[True,True,False,False]] # error