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

format

Browse Source
This commit is contained in:
Awni Hannun
2023-11-30 11:50:36 -08:00
parent 266c4e3df6
commit db487e6b1a
6 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/mlx/utils.py
View File

@@ -1,5 +1,6 @@
# Copyright © 2023 Apple Inc.
def tree_map(fn, tree, *rest):
"""Applies ``fn`` to the leaves of the python tree ``tree`` and
returns a new collection with the results.