Tensor.diag (#11108)

also updated Tensor.eye to use it
This commit is contained in:
chenyu
2025-07-05 23:03:02 -04:00
committed by GitHub
parent 4905af4ae0
commit 845a4d32bc
3 changed files with 16 additions and 1 deletions

View File

@@ -26,5 +26,6 @@
::: tinygrad.Tensor.transpose
::: tinygrad.Tensor.flatten
::: tinygrad.Tensor.unflatten
::: tinygrad.Tensor.diag
::: tinygrad.Tensor.roll
::: tinygrad.Tensor.rearrange