void SetReduceAxes(std::vector<int64_t> reduceAxes) noexcept;
函数功能
设置需要进行归约操作的轴。
函数原型
void SetReduceAxes(std::vector<int64_t> reduceAxes) noexcept;
约束说明
reduceAxes指定的轴必须在[0, inputdimsSize)的有效取值范围，inputdimsSize是AddReduce时的输入input的维数。
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
reduceAxes |
输入 |
指定轴进行归约。 |
返回值说明
无
父主题： class ReduceLayer