megengine.functional.nn.local_conv2d

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

在输入的4D张量上使用多个untied kernel进行空域卷积。它也被称为局部连接层。

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

  • weight (Tensor) – 卷积核。形状通常是 (out_channels, in_channels, height, width).

  • 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

返回

输出张量。