jax.numpy as jnp jnp.einsum('bd,df->bf', x, Win)
jax.numpy
Creator
Creator
Seonglae ChoCreated
Created
2025 Feb 6 11:37Editor
Editor
Seonglae ChoEdited
Edited
2025 Feb 6 11:38Refs
Refs
Loading views...
Seonglae ChoSeonglae ChoSeonglae ChoSeonglae Chojax.numpy as jnp jnp.einsum('bd,df->bf', x, Win)