Vector dot not equal to array dot

Maybe more a comment than an issue: I find it very confusing that the dot product of jft.Vector does not equal the dot product of jnp.array. I would expect that in the following code res_array and res_vec would be equal, but they are not.

import nifty8.re as jft
import jax.numpy as jnp

a = jnp.arange(12).reshape((3,4))
b = jnp.arange(12).reshape((4,3))
res_array = a.dot(b) # will have shape (3,3)

a_vec = jft.Vector(a)
b_vec = jft.Vector(b)
res_vec = a_vec.dot(b_vec) # will have shape ()

For arrays dot performs matrix multiplication, for vectors dot does an inner product.

@gedenhof, what do you think? Should we maybe rename the dot of vectors?

Assignee Loading
Time tracking Loading