megengine.functional.conv2d

conv2d(inp, weight, bias=None, stride=1, padding=0, dilation=1, groups=1, conv_mode='CROSS_CORRELATION', compute_mode='DEFAULT')[源代码]

二维卷积运算。

更多信息参见 Conv2d 。

参数

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

  • weight (Tensor) – 卷积核。

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

  • stride (Union[int, Tuple[int, int]]) – 二维卷积运算的步长。默认：1

  • padding (Union[int, Tuple[int, int]]) – 输入数据空域维度两侧的填充（padding）大小。仅支持填充0值。默认：0

  • dilation (Union[int, Tuple[int, int]]) – 二维卷积运算的空洞（dilation）。默认：1

  • groups (int) – 将输入和输出通道划分成的组数, 以便执行分组卷积 “grouped convolution” 。 当组数 groups 不为1时,输入通道 in_channels 和输出通道 out_channels 必须能被 groups 整除， 权值矩阵的形状应为 (groups, out_channel // groups, in_channels // groups, height, width) 。

  • conv_mode (string or Convolution.Mode) – 支持 “CROSS_CORRELATION”。默认：”CROSS_CORRELATION”

  • compute_mode (string or Convolution.ComputeMode) – 当该值设置为 ‘DEFAULT’ 时，对中间结果的精度不做特殊要求。当设置为 ‘FLOAT32’ 时，累加器和中间结果将使用 Float32 类型，但该设置仅当输入和输出为 Float16 类型时有效。

返回类型

Tensor

返回

输出张量。

megengine.functional.conv1d megengine.functional.conv3d