重点就是理解Conv1d纵列进行滑动
In [12]: a = torch.randn(20,16,50)
In [13]: a.shape
Out[13]: torch.Size([20, 16, 50])
In [14]: m = torch.nn.Conv1d(16, 33, 1)
In [15]: m(a).shape
Out[15]: torch.Size([20, 33, 50])
本文介绍了PyTorch中的Conv1d和Conv2d的区别。Conv1d主要用于一维数据,如时间序列分析,它沿着一个维度进行卷积操作。而Conv2d适用于二维数据,如图像处理,它在两个维度上进行卷积。通过实例展示了Conv1d如何保持输入和输出的宽度不变,是理解这两种卷积操作的关键。
重点就是理解Conv1d纵列进行滑动
In [12]: a = torch.randn(20,16,50)
In [13]: a.shape
Out[13]: torch.Size([20, 16, 50])
In [14]: m = torch.nn.Conv1d(16, 33, 1)
In [15]: m(a).shape
Out[15]: torch.Size([20, 33, 50])
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1万+
2万+
1万+
7216
2632

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