码迷,mamicode.com
首页 > 其他好文 > 详细

nn.ReflectionPad2d(镜像填充)

时间:2020-01-30 21:22:44      阅读:238      评论:0      收藏:0      [点我收藏+]

标签: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

nn.ReflectionPad2d(镜像填充)

标签:tor   官方文档   alt   pad   otto   ref   add   NPU   输入   

原文地址:https://www.cnblogs.com/lyp1010/p/12243732.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!