-call numpy.dot for inputs with dim == 0 or dim > 2
-add gradient calcs for various cases
-add infer_shape for various cases
-update docstring