From bfbea8fd948660abbeb31010bd165f502b344095 Mon Sep 17 00:00:00 2001 From: zhiboniu <31800336+zhiboniu@users.noreply.github.com> Date: Mon, 12 Jul 2021 11:02:02 +0800 Subject: [PATCH] add paddle/linalg.py to add new linalg apis (#34033) --- python/paddle/__init__.py | 1 + python/paddle/device.py | 2 +- python/paddle/linalg.py | 23 +++++++++++++++++++++++ 3 files changed, 25 insertions(+), 1 deletion(-) create mode 100644 python/paddle/linalg.py diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py index 773ae61a691..7490a0c29c2 100755 --- a/python/paddle/__init__.py +++ b/python/paddle/__init__.py @@ -283,6 +283,7 @@ from . import callbacks # noqa: F401 from .hapi import summary # noqa: F401 from .hapi import flops # noqa: F401 from . import hub # noqa: F401 +from . import linalg # noqa: F401 import paddle.text # noqa: F401 import paddle.vision # noqa: F401 diff --git a/python/paddle/device.py b/python/paddle/device.py index cf445917dd5..b7f69720a6e 100644 --- a/python/paddle/device.py +++ b/python/paddle/device.py @@ -22,7 +22,7 @@ from paddle.fluid.framework import is_compiled_with_cuda # noqa: F401 from paddle.fluid.framework import is_compiled_with_rocm # noqa: F401 -__all__ = [ # npqa +__all__ = [ # noqa 'get_cudnn_version', 'set_device', 'get_device', diff --git a/python/paddle/linalg.py b/python/paddle/linalg.py new file mode 100644 index 00000000000..5cef01d18ac --- /dev/null +++ b/python/paddle/linalg.py @@ -0,0 +1,23 @@ +# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +from .tensor.linalg import cholesky # noqa: F401 +from .tensor.linalg import norm # noqa: F401 +from .tensor import inverse as inv # noqa: F401 + +__all__ = [ + 'cholesky', #noqa + 'norm', + 'inv' +] -- GitLab