首页 > 其他分享> 文章详细

17 卷积操作讲解

2022-05-17 17:33:01 阅读：171 来源： 互联网

标签：kernel nn 17 卷积 torch 讲解 print input

一、卷积函数

1.二维卷积

torch.nn.functional包含在torch.nn,是更详细的方法的解说

卷积核的运算

padding运算

2. 案例

代码

import torch
import torch.nn.functional as F

# 1. 输入张量（图片）
input=torch.tensor([[1,2,0,3,1],
                    [0,1,2,3,1],
                    [1,2,1,0,0],
                    [5,2,3,1,1],
                    [2,1,0,1,1]])
# 2. 卷积核
kernel=torch.tensor([[1,2,1],
                     [0,1,0],
                     [2,1,0]])

# 3. 输出
print(input.shape)#输入的形状H,W
print(kernel.shape)#卷积核的形状H,W

# 4. 对输入和卷积进行尺寸变换，满足卷积的参数输入->H,W->batchsize,channel,H,W
input=torch.reshape(input,(1,1,5,5))
kernel=torch.reshape(kernel,(1,1,3,3))

# 4.1 输出卷积后的结果
output=F.conv2d(input,kernel,stride=1)
print(output)

# 4.2 步伐为2
output1=F.conv2d(input,kernel,stride=2)
print(output1)

# 4.3 测试padding
output2=F.conv2d(input,kernel,stride=1,padding=1)
print(output2)

运行结果

标签：kernel,nn,17,卷积,torch,讲解,print,input
来源： https://www.cnblogs.com/yuyingblogs/p/16281679.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

17 卷积操作讲解

一、卷积函数

1.二维卷积

2. 案例