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方法测试——卷积(一维)
最新推荐文章于 2024-07-28 09:26:52 发布
本文探讨了PyTorch中一维卷积的使用,通过实例展示了如何进行一维卷积计算,并得出初步的实验结果。
摘要由CSDN通过智能技术生成