未验证 提交 44d15ab1 编写于 作者: zhouweiwei2014's avatar zhouweiwei2014 提交者: GitHub

fix doc of triangular_solve (#46943)

上级 fa2f67a5
...@@ -662,6 +662,7 @@ echo ======================================== ...@@ -662,6 +662,7 @@ echo ========================================
echo Step 4. Running unit tests ... echo Step 4. Running unit tests ...
echo ======================================== echo ========================================
pip install requests
pip install -r %work_dir%\python\unittest_py\requirements.txt pip install -r %work_dir%\python\unittest_py\requirements.txt
if %ERRORLEVEL% NEQ 0 ( if %ERRORLEVEL% NEQ 0 (
echo pip install unittest requirements.txt failed! echo pip install unittest requirements.txt failed!
...@@ -678,8 +679,7 @@ dir %THIRD_PARTY_PATH:/=\%\install\zlib\bin ...@@ -678,8 +679,7 @@ dir %THIRD_PARTY_PATH:/=\%\install\zlib\bin
dir %THIRD_PARTY_PATH:/=\%\install\mklml\lib dir %THIRD_PARTY_PATH:/=\%\install\mklml\lib
dir %THIRD_PARTY_PATH:/=\%\install\mkldnn\bin dir %THIRD_PARTY_PATH:/=\%\install\mkldnn\bin
dir %THIRD_PARTY_PATH:/=\%\install\warpctc\bin dir %THIRD_PARTY_PATH:/=\%\install\warpctc\bin
dir %THIRD_PARTY_PATH:/=\%\install\onnxruntime\lib
pip install requests
set PATH=%THIRD_PARTY_PATH:/=\%\install\openblas\lib;%THIRD_PARTY_PATH:/=\%\install\openblas\bin;^ set PATH=%THIRD_PARTY_PATH:/=\%\install\openblas\lib;%THIRD_PARTY_PATH:/=\%\install\openblas\bin;^
%THIRD_PARTY_PATH:/=\%\install\zlib\bin;%THIRD_PARTY_PATH:/=\%\install\mklml\lib;^ %THIRD_PARTY_PATH:/=\%\install\zlib\bin;%THIRD_PARTY_PATH:/=\%\install\mklml\lib;^
...@@ -689,7 +689,9 @@ set PATH=%THIRD_PARTY_PATH:/=\%\install\openblas\lib;%THIRD_PARTY_PATH:/=\%\inst ...@@ -689,7 +689,9 @@ set PATH=%THIRD_PARTY_PATH:/=\%\install\openblas\lib;%THIRD_PARTY_PATH:/=\%\inst
%PATH% %PATH%
REM TODO: make ut find .dll in install\onnxruntime\lib REM TODO: make ut find .dll in install\onnxruntime\lib
xcopy %THIRD_PARTY_PATH:/=\%\install\onnxruntime\lib\onnxruntime.dll %work_dir%\%BUILD_DIR%\paddle\fluid\inference\tests\api\ /Y if "%WITH_ONNXRUNTIME%"=="ON" (
xcopy %THIRD_PARTY_PATH:/=\%\install\onnxruntime\lib\onnxruntime.dll %work_dir%\%BUILD_DIR%\paddle\fluid\inference\tests\api\ /Y
)
if "%WITH_GPU%"=="ON" ( if "%WITH_GPU%"=="ON" (
call:parallel_test_base_gpu call:parallel_test_base_gpu
......
...@@ -3000,11 +3000,21 @@ def triangular_solve(x, ...@@ -3000,11 +3000,21 @@ def triangular_solve(x,
unitriangular=False, unitriangular=False,
name=None): name=None):
r""" r"""
Computes the solution of a system of equations with a triangular coefficient matrix `x` and Computes the solution of a system of equations with a triangular coefficient. `x` is coefficient matrix
multiple right-hand sides `y` . `y` is multiple right-hand sides of equations.
Input `x` and `y` is 2D matrices or batches of 2D matrices. If the inputs are batches, the outputs Input `x` and `y` is 2D matrices or batches of 2D matrices. If the inputs are batches, the outputs is also
is also batches. batches.
Equations can be described as:
.. math::
x * Out = y
Solution of Equations is:
.. math::
Out = x ^ {-1} * y
Args: Args:
x (Tensor): The input triangular coefficient matrix. Its shape should be `[*, M, M]`, where `*` is zero or x (Tensor): The input triangular coefficient matrix. Its shape should be `[*, M, M]`, where `*` is zero or
...@@ -3031,7 +3041,6 @@ def triangular_solve(x, ...@@ -3031,7 +3041,6 @@ def triangular_solve(x,
# -x3 = 5 # -x3 = 5
import paddle import paddle
x = paddle.to_tensor([[1, 1, 1], x = paddle.to_tensor([[1, 1, 1],
[0, 2, 1], [0, 2, 1],
[0, 0,-1]], dtype="float64") [0, 0,-1]], dtype="float64")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册