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

fix doc of triangular_solve (#46943)

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