doc: manual_seed (tinygrad#4987)

there was a docstring just not linked to the doc page. also updated the example to show re-seed instead of a internal variable
Samarkhare · Jun 15, 2024 · 20b50d8 · 20b50d8
1 parent ce1ed37
commit 20b50d8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/docs/tensor/creation.md b/docs/tensor/creation.md
@@ -12,6 +12,7 @@
 
 ## Creation (random)
 
+::: tinygrad.Tensor.manual_seed
 ::: tinygrad.Tensor.rand
 ::: tinygrad.Tensor.randn
 ::: tinygrad.Tensor.randint

diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py
@@ -372,7 +372,13 @@ def manual_seed(seed=0):
 
     ```python exec="true" source="above" session="tensor" result="python"
     Tensor.manual_seed(42)
-    print(Tensor._seed)
+    print(Tensor.rand(5).numpy())
+    print(Tensor.rand(5).numpy())
+    ```
+    ```python exec="true" source="above" session="tensor" result="python"
+    Tensor.manual_seed(42)  # reset to the same seed
+    print(Tensor.rand(5).numpy())
+    print(Tensor.rand(5).numpy())
     ```
     """
     Tensor._seed, Tensor._rng_counter = seed, Tensor([0], dtype=dtypes.uint32, requires_grad=False)