import torch import torch.nn as nn m = nn.Conv1d(2, 5, 2, stride=2) print(m) input = torch.randn(2,2, 5) print(input) output = m(input) print(m.weight) print(m.bias) print(output) print(output.size()) prin
pytorch方法测试——卷积(一维)
最新推荐文章于 2025-05-23 14:30:00 发布