void SetReduceAxes(std::vector<int64_t> reduceAxes) noexcept;

函数功能

设置在哪些轴进行归约操作。

函数原型

void SetReduceAxes(std::vector<int64_t> reduceAxes) noexcept;

约束说明

参数说明

参数名

输入/输出

说明

reduceAxes

输入

指定轴进行归约。

返回值说明