zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-26 15:58:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[Issue #1187] Add nan_to_num function initial attempt (#1247)

Browse Source 
* initial attempt, working with wrong types

* not compiling; mx.float16 and mx.bfloat16 tests added

* fix nan to num

* nit

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
Anton Belov
2024-07-25 17:57:37 +01:00
committed by GitHub
parent baf9fa5f42
commit 5029894662
5 changed files with 93 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/src/python/ops.rst
View File

@@ -106,6 +106,7 @@ Operations
minimum
moveaxis
multiply
nan_to_num
negative
not_equal
ones