Add isinf and isnan ops to Tensor (#7484)

* move isinf and isnan to new branch

* sneak a roll documentation fix in

* add to docs

* update test coverage for detect_positive and detect_negative

* add types to isinf args
This commit is contained in:
geohotstan
2024-11-03 00:12:52 +08:00
committed by GitHub
parent 72a9ac27e9
commit 585f3a0f24
3 changed files with 32 additions and 2 deletions

View File

@@ -17,6 +17,8 @@ Elementwise ops operate on a per element basis. They don't change the shape of t
::: tinygrad.Tensor.ceil
::: tinygrad.Tensor.floor
::: tinygrad.Tensor.round
::: tinygrad.Tensor.isinf
::: tinygrad.Tensor.isnan
::: tinygrad.Tensor.lerp
::: tinygrad.Tensor.square
::: tinygrad.Tensor.clamp