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