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
约束说明
- reduceAxes：reduceAxes指定的轴必须在[0, inputdimsSize)有效取值范围，inputdimsSize是AddReduce的input的维数。
- operation：必须是ReduceOperation已支持的类型。
- input维度必须在[1, 8]。
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
input |
输入 |
Reduce层的输入张量。 |
operation |
输入 |
归约操作类型，取值详见enum class ReduceOperation。 |
reduceAxes |
输入 |
进行归约的轴。 例如：reduceAxes = { 0, 1 }, 表示要在0轴和1轴进行归约操作。 |
keepDimensions |
输入 |
|
返回值说明
返回一个ReduceLayer对象，如果添加失败则返回nullptr或抛出异常。
父主题： class Network