测试代码:
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)
本文主要介绍了如何在PyTorch中进行二维卷积的测试代码,虽然卷积概念普遍理解,但文中会强调一些关键细节。
测试代码:
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)
776
5346

被折叠的 条评论
为什么被折叠?