tf.identity_n

Возвращает список тензоров с такими же формами и содержимым, как у входных

tf.identity_n(
    input, name=None
)

тензоров.

Этот оператор можно использовать для переопределения градиента для сложных функций. Например, предположим, что y = f(x), и мы хотим применить пользовательскую функцию g для обратного распространения, так что dx = g(dy). В Python,

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

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