pytorch conv1d

conv1d从命名就可以看出就是一维卷积,其具体函数接口如下(该内容直接摘自官网):
pytorch conv1d_第1张图片
pytorch conv1d_第2张图片

import torch
import torch.nn as nn
m = nn.Conv1d(16,33,3,stride = 2)
input = torch.randn(20,16,50)
output = m(input)
print('output.shape = ',output.shape)

output.shape = torch.Size([20, 33, 24])

从上面理论和程序运行结果可理解如下图:
pytorch conv1d_第3张图片

你可能感兴趣的:(pytorch,pytorch,深度学习,神经网络)