nn.ReflectionPad2d()
功能:对数据进行padding
公式:
H(out) = H(in) + paddingTop + paddingBottom
W(out) = W(in) + paddingLeft + paddingRight
案例:
import torch
import torch.nn as nn
x = torch.randin((2,3,3,3))
pad = nn.ReflectionPad2d((2,2,1,1)) # left,right,top,bottom
out = pad(x) #(shape=(2,3,5,7))
测试了一下发现是以复制进行padding。
input:
out: