github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

docs: fix stack and add dtype.DType (#4721)

Browse Source
This commit is contained in:
chenyu
2024-05-24 18:23:01 -04:00
committed by GitHub
parent a16d2572a0
commit edf27470c1
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/dtypes.md
View File

@@ -1,4 +1,6 @@
::: tinygrad.dtypes ::: tinygrad.dtype.DType
::: tinygrad.dtype.dtypes
options: options:
members: true members: true
members_order: source members_order: source

2
tinygrad/tensor.py
View File

@@ -986,7 +986,7 @@ class Tensor:
print(t0.stack(t1, t2, dim=0).numpy()) print(t0.stack(t1, t2, dim=0).numpy())
``` ```
```python exec="true" source="above" session="tensor" result="python" ```python exec="true" source="above" session="tensor" result="python"
print(t0.stack([t1, t2], dim=1).numpy()) print(t0.stack(t1, t2, dim=1).numpy())
``` ```
""" """
# checks for shapes and number of dimensions delegated to cat # checks for shapes and number of dimensions delegated to cat