implement logcumsumexp (#6921)

* implement logcumsumexp

* change axis=None to axis=0
This commit is contained in:
jeffzh4ng
2024-10-06 10:45:36 -04:00
committed by GitHub
parent f588169fdc
commit 19a7e41113
3 changed files with 36 additions and 0 deletions

View File

@@ -13,6 +13,7 @@
::: tinygrad.Tensor.softmax
::: tinygrad.Tensor.log_softmax
::: tinygrad.Tensor.logsumexp
::: tinygrad.Tensor.logcumsumexp
::: tinygrad.Tensor.argmax
::: tinygrad.Tensor.argmin