Save the date! Google I/O returns May 18-20 Register now


TensorFlow 1 version

Returns a list of tensors with the same shapes and contents as the input


This op can be used to override the gradient for complicated functions. For example, suppose y = f(x) and we wish to apply a custom function g for backprop such that dx = g(dy). In Python,

with tf.get_default_graph().gradient_override_map(
    {'IdentityN': 'OverrideGradientWithG'}):
  y, _ = identity_n([f(x), x])

def ApplyG(op, dy, _):
  return [None, g(dy)]  # Do not backprop to f(x).

input A list of Tensor objects.
name A name for the operation (optional).

A list of Tensor objects. Has the same type as input.