megengine.functional.nn.conv1d

conv1d(inp, weight, bias=None, stride=1, padding=0, dilation=1, groups=1, conv_mode='cross_correlation', compute_mode='default')[源代码]

一维卷积运算。

更多信息参见 Conv1d

参数
  • inp (Tensor) – 卷积运算的特征图。

  • weight (Tensor) – 卷积核。

  • bias (Optional[Tensor]) – 添加到卷积结果中的偏置量(如果给定该值)

  • stride (int) – 一维卷积运算中的步长。 默认: 1

  • padding (int) – 在输入值的空间维度上每一侧填充的尺寸。 仅支持用零值填充。 默认: 0

  • dilation (int) – 一维卷积运算的扩张值(dilation)。 默认: 1

  • groups (int) – 在进行“分组卷积”时,对输入输出通道的分组数量。当 groups 不是 1 时,in_channelsout_channels 必须可被 groups 整除,卷积核权重的形状将会是 (groups, out_channel // groups, in_channels // groups, kernel_size)。默认值:1

  • conv_mode – 仅支持 ‘cross_correlation’。默认值:’cross_correlation’

  • compute_mode – 设置为 default 时,计算时中间结果的精度不会有特殊的保证;设置为 float32 时,计算时累加器和中间结果的类型会设为 float32,仅当输入和输出张量的数据类型都是 float16 时生效。

返回类型

Tensor