mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #26245 from jakevdp:fix-line

Browse Source 
PiperOrigin-RevId: 721879951
This commit is contained in:
jax authors 2025-01-31 13:16:51 -08:00
commit 70aed64a15
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/_src/numpy/lax_numpy.py
View File

@ -13407,7 +13407,6 @@ def place(arr: ArrayLike, mask: ArrayLike, vals: ArrayLike, *,
"""
data, mask_arr, vals_arr = util.ensure_arraylike("place", arr, mask, vals)
vals_arr = vals_arr.ravel()
data, mask_arr, vals_arr = asarray(arr), asarray(mask), ravel(vals)
if inplace:
raise ValueError(
"jax.numpy.place cannot modify arrays in-place, because JAX arrays are immutable. "