mwitiderrick/update_jax_array.py

Created July 11, 2022 09:11

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/mwitiderrick/a32641acd149ad0a28696bf488240141.js"></script>
Save mwitiderrick/a32641acd149ad0a28696bf488240141 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

update_jax_array.py

	new_scores_array = scores_array.at[0:3].set([20,40,90])
	new_scores_array
	# DeviceArray([20, 40, 90, 30, 25], dtype=int32)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment