github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

docs: update wording for unflatten (#4974)

Browse Source 
it was using `Expands`, the same in torch doc, but we also have expand so it's confusing
This commit is contained in:
chenyu
2024-06-14 23:12:41 -04:00
committed by GitHub
parent efbf4fca05
commit 5f7dd74655
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/tensor.py
View File

@@ -1228,7 +1228,7 @@ class Tensor:
def unflatten(self, dim:int, sizes:Tuple[int,...]):
"""
Expands dimension `dim` of the tensor over multiple dimensions specified by `sizes`.
Unflattens dimension `dim` of the tensor into multiple dimensions specified by `sizes`. `Tensor.flatten()` is the inverse of this function.
```python exec="true" source="above" session="tensor" result="python"
print(Tensor.ones(3, 4, 1).unflatten(1, (2, 2)).shape)