标签:tor 官方文档 alt pad otto ref add NPU 输入
1、官方文档上 是这个
2、输出是
H(out) = H(in) + paddingTop + paddingBottom
W(out) = W(in) + paddingLeft + paddingRight
直接来例子试试 最重要的是下面 # laft, right, top, bottom 这个注释
input = torch.randn(64, 3, 220, 220) # input size
# 输入4-tuple
pad = nn.ReflectionPad2d((3, 3, 5, 5)) # laft, right, top, bottom
output = pad(input) # size(64, 3, 230, 226)
# 输入int
pad = nn.ReflectionPad2d(3)
output = pad(input) # size(64, 3, 226, 226)
通过运算 发现的确是这样的结果 所以记住公式就行了
转自:https://blog.csdn.net/bc521bc/article/details/85134539
标签:tor 官方文档 alt pad otto ref add NPU 输入
原文地址:https://www.cnblogs.com/lyp1010/p/12243732.html