megengine.functional.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) – 一维卷积运算中的步长。 默认： 1padding (
int) – 在输入值的空间维度上每一侧填充的尺寸。 仅支持用零值填充。 默认： 0dilation (
int) – 一维卷积运算的扩张值(dilation)。 默认： 1groups (
int) – 将输入和输出通道划分成的组数, 以便执行分组卷积 “grouped convolution” 。 当组数groups不为1时,输入通道in_channels和输出通道out_channels必须能被groups整除， 权值矩阵的形状应为 (groups, out_channel // groups, in_channels // groups, height, width) 。conv_mode (string or
mgb.opr_param_defs.Convolution.Mode) – 支持 ‘CROSS_CORRELATION’。默认：’CROSS_CORRELATION’。compute_mode (string or
mgb.opr_param_defs.Convolution.ComputeMode) – 当该值设置为 ‘DEFAULT’ 时，对中间结果的精度不做特殊要求。当设置为 ‘FLOAT32’ 时，累加器和中间结果将使用 Float32 类型，但该设置仅当输入和输出为 Float16 类型时有效。
- 返回类型
Tensor