megengine.functional.quantized.conv_bias_activation

conv_bias_activation(inp, weight, bias, dtype=None, stride=1, padding=0, dilation=1, groups=1, nonlinear_mode='identity', conv_mode='cross_correlation', compute_mode='default')

带有激活函数的卷积偏置操作，仅用于推理

参数

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

• weight (Tensor) – 卷积核。

• bias (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) – 输入和输出通道被分成的组数，以便执行分组卷积。当 groups 不为1时，in_channels 和 out_channels 必须被 groups 整除，权重的形状为 (groups, out_channel // groups, in_channels // groups, height, width).

• conv_mode – 支持 ‘cross_correlation’ 或者 ‘convolution’. 默认：’cross_correlation’

• dtype – 支持 np.dtype, 默认是 np.int8

• compute_mode – 设置为 default 时，计算时中间结果的精度不会有特殊的保证；设置为 float32 时，计算时累加器和中间结果的类型会设为 float32，仅当输入和输出张量的数据类型都是 float16 时生效。

返回类型

Tensor

上一页

megengine.functional.nn.nvof

下一页

megengine.functional.quantized.batch_conv_bias_activation