tf.raw_ops.IdentityN

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

tf.raw_ops.IdentityN(
    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).