Compute the dot product of two or more arrays in a single function call,
while automatically selecting the fastest evaluation order.
If the first argument is 1D, it is treated as a row vector. If the last
argument is 1D, it is a column vector. All other intermediate arguments
must be 2D matrices.
Equivalent to jax.numpy.einsum() with the appropriate subscripts.
Compute the dot product of two or more arrays in a single function call, while automatically selecting the fastest evaluation order.
If the first argument is 1D, it is treated as a row vector. If the last argument is 1D, it is a column vector. All other intermediate arguments must be 2D matrices.
Equivalent to
jax.numpy.einsum()with the appropriate subscripts.