测试代码:
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(
测试代码:
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(