ReduceOperation

功能

在指定维度上求和、取最大值或最小值，并消除这个维度。

定义

struct ReduceParam {
    enum ReduceType {
        REDUCE_UNDEFINED = 0,
        REDUCE_MAX,
        REDUCE_MIN,
        REDUCE_SUM,
    };
    ReduceType reduceType = REDUCE_UNDEFINED;
    SVector<int64_t> axis;
};

成员

成员名称	描述
ReduceType	计算类型，支持以下参数： REDUCE_UNDEFINED：未定义。 REDUCE_MAX：求最大值。 REDUCE_MIN：求最小值。 REDUCE_SUM：求和。
axis	指定轴（维度）。 axis不能为空且长度要求小于等于输入x的维度。 axis可以支持多个轴上进行处理，各元素要求小于x的维度且大于等于0。

成员名称

描述

ReduceType

计算类型，支持以下参数：

REDUCE_UNDEFINED：未定义。
REDUCE_MAX：求最大值。
REDUCE_MIN：求最小值。
REDUCE_SUM：求和。

axis

指定轴（维度）。

axis不能为空且长度要求小于等于输入x的维度。
axis可以支持多个轴上进行处理，各元素要求小于x的维度且大于等于0。

输入

参数	维度	数据类型	格式	描述
x	[-1,…,-1] -1表示当前维度的大小没有约束。	REDUCE_MAX,REDUCE_MIN:int32/REDUCE_SUM:float16	ND	输入tensor。

参数

维度

数据类型

格式

描述

[-1,…,-1]

-1表示当前维度的大小没有约束。

REDUCE_MAX,REDUCE_MIN:int32/REDUCE_SUM:float16

输入tensor。

输出

参数	维度	数据类型	格式	描述
output	基于输入“x”的维度，消除axis要求的维度。	REDUCE_MAX,REDUCE_MIN:int32/REDUCE_SUM:float16	ND	输出tensor。

父主题： atb/infer_op_params.h