当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.raw_ops.IdentityN用法及代码示例


返回与输入具有相同形状和内容的张量列表

用法

tf.raw_ops.IdentityN(
    input, name=None
)

参数

  • input Tensor 对象的列表。
  • name 操作的名称(可选)。

返回

  • Tensor 对象的列表。具有与 input 相同的类型。

张量。

此操作可用于覆盖复杂函数的梯度。例如，假设 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).

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.raw_ops.IdentityN。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。