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).