added keepdimsto ensure dimensions are kept when summing
einsum victory yay