鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

功能描述

算子功能

Reduction算子是在tensor的一个或者多个维度对每个目标元素进行指定的算法操作,数学表达式为:,其中reduce_op包含如表1 reduce_op算法操作所示的算法操作。
表1 reduce_op算法操作

reduce_op

功能

max

获取Tensor在规约维度上的最大值。

min

获取Tensor在规约维度上的最小值。

sum

获取Tensor在规约维度上的元素加和。

mul

获取Tensor在规约维度上的元素乘积。

mean

获取Tensor在规约维度上的元素平均值。