[PyTorch] masked_fill() 함수 사용 예시
파이토치에서 masked_fill() 함수는 파이토치 tensor 의 특정 값을 다른 값으로 바꾸고자 할 때 사용된다. document 가 좀 설명이 부실한 것 같아서 예시로 하나 사용해보면 import torch import torch.nn.functional as F example = torch.FloatTensor([[1, 2, 3], [4, 5, 6]]) # 특정 행렬 mask = torch.FloatTensor([[0, 1, 1], [1, 0, 1]]) # 바꾸고자 하는 행렬의 원소를 0 고정하고자 하는 행렬의 원소를 1로 둔다. result = example.masked_fill(mask == 0, 5.5) # 0을 5.5로 변환 print(result)
2023. 1. 2.
