在网络中添加一个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 |
输入 |
|
返回一个ReduceLayer对象,如果添加失败时返回nullptr。