update README (#1078)

Author: HydrogenSulfate

README.md

@@ -247,19 +247,17 @@ python -c "import paddle; paddle.utils.run_check()"
 
 | 工具 | 简介 | 支持情况 |
 | -- | -- | -- |
-| [Modulus-sym](https://github.com/PaddlePaddle/modulus-sym/tree/paddle?tab=readme-ov-file#modulus-symbolic-betapaddle-backend) | AI仿真套件 | 全量支持 |
 | [DeepXDE](https://github.com/lululxvi/deepxde/tree/master?tab=readme-ov-file#deepxde) | 方程求解套件 | 全量支持 |
 | [DeepMD-kit](https://docs.deepmodeling.com/projects/deepmd/en/latest/index.html) | 分子动力学套件 | 部分支持 |
-| [TensorLy](https://tensorly.org/dev/index.html) | 张量计算库 | 全量支持 |
-| [NVIDIA/warp](https://github.com/NVIDIA/warp) | 高性能仿真/图形库 | 全量支持 |
-| [DLPACK(v0.8)](https://dmlc.github.io/dlpack/latest/index.html) | 跨框架张量内存共享协议 | 全量支持 |
-| [tensorly](https://dmlc.github.io/dlpack/latest/index.html) | 张量运算库 | 全量支持 |
-| [warp](https://dmlc.github.io/dlpack/latest/index.html) | 基于 Python 的 GPU 高性能仿真和图形库 | 全量支持 |
+| [Modulus-sym](https://github.com/PaddlePaddle/modulus-sym/tree/paddle?tab=readme-ov-file#modulus-symbolic-betapaddle-backend) | AI仿真套件 | 全量支持 |
+| [NVIDIA/warp](https://github.com/NVIDIA/warp) | 基于 Python 的 GPU 高性能仿真和图形库 | 全量支持 |
+| [tensorly](https://github.com/tensorly/tensorly) | 张量运算库 | 全量支持 |
 | [Open3D](https://github.com/PFCCLab/Open3D.git) | 三维图形库 | 全量支持 |
+| [neuraloperator](https://github.com/PFCCLab/neuraloperator) | 神经算子库 | 全量支持 |
 | [paddle_scatter](https://github.com/PFCCLab/paddle_scatter) | 张量稀疏计算库 | 全量支持 |
+| [paddle_harmonics](https://github.com/PFCCLab/paddle_harmonics.git) | 球面谐波变换库 | 全量支持 |
 | [deepali](https://github.com/PFCCLab/deepali) | 图像、点云配准库 | 全量支持 |
-| [paddle_harmonics](https://dmlc.github.io/dlpack/latest/index.html) | 球面谐波变换库 | 全量支持 |
-| [neuraloperator](https://dmlc.github.io/dlpack/latest/index.html) | 神经算子库 | 全量支持 |
+| [DLPACK(v0.8)](https://dmlc.github.io/dlpack/latest/index.html) | 跨框架张量内存共享协议 | 全量支持 |
 <!-- --8<-- [end:adaptation] -->
 
 <!-- --8<-- [start:support] -->

ppsci/equation/pde/helmholtz.py

@@ -54,12 +54,14 @@ class Helmholtz(base.PDE):
 
     Args:
         dim (int): Dimension of equation.
+        k (float): The wave number, which is a parameter that affects the frequency of the solution.
         detach_keys (Optional[Tuple[str, ...]]): Keys used for detach during computing.
             Defaults to None.
 
     Examples:
         >>> import ppsci
-        >>> pde = ppsci.equation.Helmholtz(2, -1.0, 1.0)
+        >>> model = ppsci.arch.MLP(("x", "y"), ("u",), 2, 32)
+        >>> pde = ppsci.equation.Helmholtz(2, -1.0, model)
     """
 
     def __init__(
@@ -94,7 +96,7 @@ def helmholtz(data_dict: Dict[str, paddle.Tensor]) -> paddle.Tensor:
                     lambda y_: self.model.forward_tensor(xs[0], y_), (xs[1],)
                 )
                 out = (self.k**2) * data_dict["u"] + u__x__x + u__y__y
-            elif self.dim >= 3:
+            elif self.dim == 3:
                 u__x__x = hvp_revrev(
                     lambda x_: self.model.forward_tensor(x_, xs[1], xs[2]), (xs[0],)
                 )