[Pytorch] torch.nn.functional.pad

torch.nn.functional.pad 공식문서

함수 형태

torch.nn.functional.pad(input, pad, mode='constant', value=None) → Tensor

패딩을 넣을 때는 좌,우,상,하,앞,뒤 로 넣을 수 있다.

t4d = torch.empty(3, 3, 4, 2)
p1d = (1, 1)

out = F.pad(t4d, p1d, "constant", 0)

예를 들어 위 코드는 입력 텐서의 마지막 차원에 좌, 우에 패딩을 1개씩 넣는 코드이다. 결과는 다음과 같다.

output:
torch.Size([3, 3, 4, 4])

좌,우,상,하에 값을 넣으려면 다음처럼 작성한다

t4d = torch.empty(3, 3, 4, 2)
p1d = (1, 1, 2, 2)

out = F.pad(t4d, p1d, "constant", 0)
print(out.size())

output:
torch.Size([3, 3, 8, 4])