(*args, **kwargs)
return backend()['custom_grad'](*args, **kwargs)
Set a custom gradient computation (override the default) for a function.
Set a custom gradient computation (override the default) for a function.
Set a custom gradient computation ( override the default ) for a function .
def custom_grad(*args, **kwargs):
  """Set a custom gradient computation (override the default) for a function."""
  return backend()['custom_grad'](*args, **kwargs)
def custom_grad ( * args , * * kwargs ) : return backend ( ) [ 'custom_grad' ] ( * args , * * kwargs )