[ONNX] add OP:erf

164d8ce5 · Channingss · 5cd17103 · 164d8ce5
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

x2paddle/op_mapper/onnx_directly_map.py x2paddle/op_mapper/onnx_directly_map.py +3 -2

未找到文件。
--- 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'],