docsfix: nn.Tensor -> Tensor

2024-06-12 09:18:32 +02:00
parent 3d13c23bfa
commit ca4ccddcd6
1 changed files with 1 additions and 1 deletions
@@ -40,7 +40,7 @@ def safe_save(tensors:Dict[str, Tensor], fn:str, metadata:Optional[Dict[str, Any
  Saves a state_dict to disk in a .safetensor file with optional metadata.

  ```python
-  t = nn.Tensor([1, 2, 3])
+  t = Tensor([1, 2, 3])
  nn.state.safe_save({'t':t}, "test.safetensor")
  ```
  """