Skip to content

Instantly share code, notes, and snippets.

@mwitiderrick

mwitiderrick/indexing.py

Created July 11, 2022 12:42
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mwitiderrick/6c410c456c367a232cbc65c331ed8f05 to your computer and use it in GitHub Desktop.
Save mwitiderrick/6c410c456c367a232cbc65c331ed8f05 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
indexing.py
matrix = jnp.arange(1,17)
matrix[20]
# DeviceArray(16, dtype=int32)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment