jax
8fc1332e
- Avoid broadcasting for batch dimensions in jnp.matmul.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Avoid broadcasting for batch dimensions in jnp.matmul. Instead, squeeze size 1 dimensions out of the matmul input, and transpose any non-batch non-contracting dimensions into the correct location.
References
#3783 - Avoid broadcasts in implementation of jnp.matmul
Author
hawkinsp
Parents
943c4216
Loading