update comments

lululxvi · Jan 3, 2025 · ab5547f · ab5547f
1 parent 150150a
commit ab5547f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepxde/gradients/gradients.py b/deepxde/gradients/gradients.py
@@ -19,7 +19,7 @@ def jacobian(ys, xs, i=None, j=None):
       computation.
 
     Args:
-        ys: Output Tensor of shape (batch_size, dim_y).
+        ys: Output Tensor of shape (batch_size, dim_y) or (batch_size1, batch_size2, dim_y).
         xs: Input Tensor of shape (batch_size, dim_x).
         i (int or None): `i`th row. If `i` is ``None``, returns the `j`th column
             J[:, `j`].