elixir-nx / nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir
2.66k stars 194 forks source link

Remove Nx.map/2 #1492

Closed jonatanklosko closed 6 months ago

jonatanklosko commented 6 months ago

It didn't prove useful at all, given that it works with scalars, it should basically always be possible to use a combination element-wise operations. Some backends don't even support it. Sidenote: I'm pretty sure jax doesn't use stablehlo.map at all.