ReduceLayer *AddReduce(Tensor *input, ReduceOperation operation, std::vector<int64_t> reduceAxes, bool keepDimensions) noexcept;

函数功能

在网络中添加一个Reduce层。

函数原型

ReduceLayer *AddReduce(Tensor *input, ReduceOperation operation, std::vector<int64_t> reduceAxes, bool keepDimensions) noexcept

约束说明

参数说明

参数名

输入/输出

说明

input

输入

Reduce层的输入张量。

operation

输入

归约操作类型,取值详见enum class ReduceOperation

reduceAxes

输入

进行归约的轴。

例如:reduceAxes = { 0, 1 }, 表示要在0轴和1轴进行归约操作。

keepDimensions

输入

  • true:保留归约操作的轴。
  • false:不保留归约操作的轴。

返回值说明

返回一个ReduceLayer对象,如果添加失败时返回nullptr。