Merge pull request #244 from Channingss/develop

[ONNX] add OP:erf

Merge pull request #244 from Channingss/develop
[ONNX] add OP:erf
b95d36f0 · Jason · GitHub · 6c3cb861 · 164d8ce5 · b95d36f0
显示空白变更内容
内联并排

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'],