더북(TheBook)

코드를 실행하면 다음과 같이 출력됩니다.

tensor(0.7357)
tensor(0.7357)
------------
tensor([[-0.3894, 0.6999, -0.6733],
        [ 1.7850, 0.1961,  0.2701]])
tensor([[-0.3894, 1.7850],
        [ 0.6999, 0.1961],
        [-0.6733, 0.2701]])

③ 텐서를 비교하는 함수로, 텐서가 서로 같은지를 비교한다면 torch.eq, 다른지를 비교한다면 torch.ne, 크거나 같은지를 비교한다면 torch.ge를 사용합니다. torch.eq는 다음과 같은 형식을 이용하여 사용합니다.

torch.eq(비교 대상 텐서, 비교할 텐서)

torch.eq의 결과는 각 텐서의 요소들을 비교해서 같으면 True, 다르면 False를 반환합니다.

예를 들어 다음과 같이 사용합니다.

torch.eq(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]]))

1과 1을 비교했을 때 서로 같으므로 True, 2와 1을 비교했을 때 서로 다르므로 False, 3과 4를 비교했을 때 서로 다르므로 False, 4와 4를 비교했을 때 서로 같으므로 True를 반환합니다. 코드를 실행하면 다음과 같이 출력될 것입니다.

tensor([[ True, False],
        [False,  True]])
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.