当前位置：网站首页>【Pytorch】torch.nn.functional.conv2d（F.conv2d） same padding实现方法（输入与输出大小相同）

【Pytorch】torch.nn.functional.conv2d（F.conv2d） same padding实现方法（输入与输出大小相同）

2022-08-06 05:26:00 【小丫么小阿豪】

Motivation

在做图像退化的时候发现pytorch早期版本不支持same padding（新版本是有padding = 'same'这个选项的，但是本人试了也没成功，暂时没找到原因）。于是去查有没有办法自己实现一下same padding。

torch.nn.functional.conv2d 详解

官方文档写的非常清楚，直接贴上来
在这里插入图片描述

实现方法

由于F.conv2d的padding最多只支持两个参数（对称填充），有些输出大小没办法实现。因此可以通过F.pad()先做填充，再用F.conv2d做卷积。

import torch.nn.functional as F
import torch.nn as nn

filters = torch.randn(1, 1, 8, 15)
inputs = torch.randn(16, 1, 256, 256)
# 手动填充
inputs = F.pad(inputs, (7, 7, 4, 3))
out = F.conv2d(inputs, filters)
print(out.shape)

输出：

torch.Size([16, 1, 256, 256])
原网站

版权声明
本文为[小丫么小阿豪]所创，转载请带上原文链接，感谢
https://blog.csdn.net/qq_43426078/article/details/124025439

边栏推荐

猜你喜欢

随机推荐