Clear field dim in docs

taichi-dev · Jan 12, 2023 · 719176d · 719176d
1 parent 731d454
commit 719176d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/lang/articles/get-started/accelerate_pytorch.md b/docs/lang/articles/get-started/accelerate_pytorch.md
@@ -180,9 +180,9 @@ The Taichi reference code is almost identical to its Python counterpart. And a g
 ```python
 @ti.kernel
 def taichi_forward_v0(
-        out: ti.types.ndarray(field_dim=3),
-        w: ti.types.ndarray(field_dim=3),
-        k: ti.types.ndarray(field_dim=3),
+        out: ti.types.ndarray(ndim=3),
+        w: ti.types.ndarray(ndim=3),
+        k: ti.types.ndarray(ndim=3),
         eps: ti.f32):
 
     for b, c, t in out: