megengine.functional.nn.conv3d

conv3d(inp, weight, bias=None, stride=1, padding=0, dilation=1, groups=1, conv_mode='cross_correlation')

三维卷积运算。

更多信息参见 Conv3d 。

参数

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

• weight (Tensor) – 卷积核。

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

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

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

• dilation (Union[int, Tuple[int, int, int]]) – 三维卷积运算的扩张值(dilation)。 默认： 1

• groups (int) – number of groups into which the input and output channels are divided, so as to perform a grouped convolution. When groups is not 1, in_channels and out_channels must be divisible by groups, and the shape of weight should be (groups, out_channel // groups, in_channels // groups, depth, height, width). Default: 1

• conv_mode (str) – supports “cross_correlation”. Default: “cross_correlation”

返回类型

Tensor

返回

输出张量。

上一页

megengine.functional.nn.conv2d

下一页

megengine.functional.nn.local_conv2d