[PyTorch] masked_fill() 함수 사용 예시

 

파이토치에서 masked_fill() 함수는 파이토치 tensor 의 특정 값을 다른 값으로 바꾸고자 할 때 사용된다.

 

공식문서에 나온 masked_fill 함수

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)