# Compute gradients
    loss.backward()