코드를 실행하면 다음과 같이 출력됩니다.
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]])