From edc2430adfcf234e454ca8a07083ea4a4f58bbfe Mon Sep 17 00:00:00 2001 From: root Date: Fri, 22 Nov 2019 03:33:34 +0000 Subject: [PATCH] fix bug for op:NonZero --- x2paddle/op_mapper/onnx_op_mapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/onnx_op_mapper.py b/x2paddle/op_mapper/onnx_op_mapper.py index ea670d7..2c9d48b 100644 --- a/x2paddle/op_mapper/onnx_op_mapper.py +++ b/x2paddle/op_mapper/onnx_op_mapper.py @@ -1103,7 +1103,7 @@ class ONNXOpMapper(OpMapper): val_x = self.graph.get_input_node(node, idx=0, copy=True) where_name = node.layer_name + '_where' node.fluid_code.add_layer("where", - inputs=val_x.layer_name + '==1', + inputs=val_x.layer_name + '!=0', output=where_name) dims = len(val_x.out_shapes[0]) elements_count_val_x = reduce(lambda x, y: x * y, val_x.out_shapes[0]) -- GitLab