General dot product/contraction operator.
Prefer higher-level functions like jax.numpy.dot(), jax.numpy.matmul(), jax.numpy.tensordot(), and jax.numpy.einsum()` where possible.
jax.numpy.dot()
jax.numpy.matmul()
jax.numpy.tensordot(), and
General dot product/contraction operator.
Prefer higher-level functions like
jax.numpy.dot(),jax.numpy.matmul(),jax.numpy.tensordot(), andjax.numpy.einsum()` where possible.