megengine.functional.quantized.batch_conv_bias_activation

batch_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.quantized.conv_bias_activation

下一页

megengine.functional.quantized.conv_transpose2d