Function save_to

imperative/python/test/unit/core/test_autodiff.py:42–46

(self, name="grad")

def save_to(self, name="grad"):
    def callback(grad):
        setattr(self, name, grad)

    return callback


@pytest.mark.require_ngpu(2)