Skip to content

Instantly share code, notes, and snippets.

@imirzadeh

imirzadeh/gist:790eb652fe9ce2b5beb3a2ab69bc64bf

Created August 20, 2021 05:30
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 imirzadeh/790eb652fe9ce2b5beb3a2ab69bc64bf to your computer and use it in GitHub Desktop.
Save imirzadeh/790eb652fe9ce2b5beb3a2ab69bc64bf to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gistfile1.txt
inp = jnp.array([[[l[x], G[i]][0] for i in range(len(G))] for x in range(len(l))])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment