MCPcopy Create free account
hub / github.com/ml-explore/mlx / random_take

Function random_take

benchmarks/python/single_ops.py:98–99  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

96 mx.eval(ids)
97
98 def random_take():
99 return [mx.take(a, idx, 0) for idx in ids]
100
101 time_fn(random_take)
102

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected