megengine.functional.local_conv2d

local_conv2d(inp, weight, bias=None, stride=1, padding=0, dilation=1, conv_mode='CROSS_CORRELATION')

使用untied kernels对图像进行二维空域卷积。

参数

• inp (megengine.tensor.Tensor) –

• weight (megengine.tensor.Tensor) –

• bias (Optional[megengine.tensor.Tensor]) –

• stride (Union[int, Tuple[int, int]]) –

• padding (Union[int, Tuple[int, int]]) –

• dilation (Union[int, Tuple[int, int]]) –

megengine.functional.conv3d megengine.functional.conv_transpose2d