Hi,
The argument grad_output is the gradient with respect to the output given by the autograd engine.
grad_input is the tensor where the result should be written.
grad_columns is a temporary buffer given here for efficiency.
The backward functions always take grad_output and return grad_input.