def add_reduce(self: Network, input: Tensor, operation: ReduceOperation, reduceAxes: int, keepDimensions: bool) -> ReduceLayer

函数功能

在网络中添加一个Reduce层。

函数原型

add_reduce(self: ascendie.pyascendie.Network, arg0: ascendie.pyascendie.Tensor, arg1: ascendie.pyascendie.ReduceOperation, arg2: int, arg3: bool) -> AscendIE::ReduceLayer

约束说明

• reduceAxes：指定的轴必须在[0, input.ndims)有效取值范围。
• operation：必须是ReduceOperation已支持的类型。
• 当前不支持动态shape。

参数说明