Python make_reduction_op示例

示例#1

文件： ops_bridge.py 项目： rsumner31/ngraph

def ReduceProd(onnx_node,
               ng_inputs):  # type: (NodeWrapper, List[TensorOp]) -> Op
    return make_reduction_op(ng.prod, onnx_node, ng_inputs[0])

示例#2

文件： ops_bridge.py 项目： rsumner31/ngraph

def ArgMax(onnx_node, ng_inputs):  # type: (NodeWrapper, List[TensorOp]) -> Op
    return make_reduction_op(ng.argmax, onnx_node, ng_inputs[0])

示例#3

文件： ops_bridge.py 项目： rsumner31/ngraph

def ReduceLogSumExp(onnx_node,
                    ng_inputs):  # type: (NodeWrapper, List[TensorOp]) -> Op
    op = ng.exp(ng_inputs[0])
    op = make_reduction_op(ng.sum, onnx_node, op)
    op = ng.log(op)
    return op