diff --git a/x2paddle/op_mapper/onnx_directly_map.py b/x2paddle/op_mapper/onnx_directly_map.py index 975eadf51cb6735b2d345e89fea2726ac6b9149a..a152bddf5b6410f20ab939259747a5c6d9dec65c 100644 --- a/x2paddle/op_mapper/onnx_directly_map.py +++ b/x2paddle/op_mapper/onnx_directly_map.py @@ -32,11 +32,12 @@ default_op_mapping = { dict(), dict( min=(_np.asarray([255, 255, 127, 255], - dtype=_np.uint8).view(_np.float32)), + dtype=_np.uint8).view(_np.float32)[0]), max=(_np.asarray([255, 255, 127, 127], - dtype=_np.uint8).view(_np.float32)), + dtype=_np.uint8).view(_np.float32)[0]), ) ], + 'Erf': ['erf', ['X'], ['Out']], 'Ceil': ['ceil', ['X'], ['Out']], 'ReduceMean': [ 'reduce_mean', ['X'], ['Out'],