一维卷积神经网络nn.Conv1d

tensorflow中的解释:

Convolution1D(nb_filter, filter_length, strides, input_shape)
-nb_filter : 卷积核的数量,也是输出的维度
-filter_length : 每个过滤器的长度
-strides: 卷积核移动的步距
-input_shape:输入信号的格式

pyotrch中的解释:

Conv1d(in_channels, out_channels, kernel_size,stride=1, padding=0, dilation=1, groups=1, bias=True)

  • in_channels(int)—输入数据的通道数。在文本分类中,即为句子中单个词的词向量的维度。 (word_vector_num)
  • out_channels(int)—输出数据的通道数。设置 N 个输出通道数,就有 N 个1维卷积核。(new word_vector_num)
  • kernel_size(int or tuple) —卷积核的长度,1维卷积中卷积核的实际大小维度是(in_channels,kernel_size),顺序不可互换。
  • stride(int or tuple, optional)—卷积步长。
  • padding (int or tuple, optional)—输入的每一条边补充0的层数。

Conv1d模型的卷积核大小是[输入通道数,卷积核的长]

一言顶过千万句


pytorch是横着做卷积的 ,TensorFlow是竖着卷积 !!!

你可能感兴趣的:(cnn,深度学习,人工智能)