功能

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

定义

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

成员

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

成员名称

描述

reduceType

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

REDUCE_MAX：求最大值。
REDUCE_MIN：求最小值。
REDUCE_SUM：求和。

axis

指定轴（维度）。

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

输入

参数	维度	数据类型	格式
x	[-1,…,-1] -1表示当前维度的大小没有约束。	int32	ND

参数

维度

数据类型

格式

x

[-1,…,-1]

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

int32

ND

输出

参数	维度	数据类型	格式
y	基于输入“x”的维度，消除axis要求的维度。	int32	ND

ReduceOperation

功能

定义

成员

输入

输出