Function run

examples/pytorch/linear_backward/reference.py:13–16

(grad_output, x, weight)

@torch.no_grad()
def run(grad_output, x, weight):
    grad_input = grad_output @ weight
    grad_weight = grad_output.mT @ x
    return grad_input, grad_weight