item assignment
#11035
-
|
In numpy, a[:, :, 0] += self.b[0] - 1
a[:, :, 1] *= 2 * self.b[1] - 1how to do this in jax? |
Beta Was this translation helpful? Give feedback.
Answered by
jakevdp
Jun 9, 2022
Replies: 1 comment
-
|
This is covered at https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#in-place-updates. Please let me know if that answers your question! |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
pure-rgb
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
This is covered at https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#in-place-updates. Please let me know if that answers your question!