fixes a bug in einsum
check that the argument axis length corresponding to the same output axis matches with the previously recorded one.
check that the argument axis length corresponding to the same output axis matches with the previously recorded one.