當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.identity_n用法及代碼示例


返回與輸入具有相同形狀和內容的張量列表

用法

tf.identity_n(
    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.identity_n。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。