【Hackathon No.25】为 Paddle 新增 nanquantile 数学计算API (#41343)

PaddlePaddle · Apr 15, 2022 · b9ee6a2 · b9ee6a2
1 parent 2eac4db
commit b9ee6a2
Show file tree

Hide file tree

Showing 5 changed files with 455 additions and 304 deletions.
diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py
@@ -329,6 +329,7 @@
 from .tensor.stat import numel  # noqa: F401
 from .tensor.stat import median  # noqa: F401
 from .tensor.stat import quantile  # noqa: F401
+from .tensor.stat import nanquantile  # noqa: F401
 from .device import get_cudnn_version  # noqa: F401
 from .device import set_device  # noqa: F401
 from .device import get_device  # noqa: F401
@@ -495,6 +496,7 @@
            'numel',
            'median',
            'quantile',
+           'nanquantile',
            'no_grad',
            'set_grad_enabled',
            'is_grad_enabled',

diff --git a/python/paddle/fluid/tests/unittests/test_quantile.py b/python/paddle/fluid/tests/unittests/test_quantile.py