An API opinion I just nonticed. I think we should not return the operand. `ufunc.at` is always in-place, so returning the operand is not the usual python design.