pytorch方法测试——卷积（二维）

tmk_01

于 2018-06-12 21:40:48 发布

阅读量2.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/tmk_01/article/details/80671335

本文主要介绍了如何在PyTorch中进行二维卷积的测试代码，虽然卷积概念普遍理解，但文中会强调一些关键细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

测试代码：

import torch
import torch.nn as nn

m = nn.Conv2d(2, 2, 3, stride=2)
input = torch.randn(1, 2, 5, 7)
output = m(input)

print("输入图片(2张)：")
print(input)
print("卷积的权重：")
print(m.weight)
print("卷积的偏重：")
print(m.bias)

最低0.47元/天解锁文章

200万优质内容无限畅学