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


Python tf.identity用法及代碼示例


返回一個與輸入具有相同形狀和內容的張量。

用法

tf.identity(
    input, name=None
)

參數

  • input TensorVariableCompositeTensor 或任何可以使用 tf.convert_to_tensor 轉換為張量的東西。
  • name 操作的名稱(可選)。

返回

  • Tensor 或 CompositeTensor。具有與 input 相同的類型和內容。

返回值與原始的 Tensor 不同,但包含相同的值。此操作在同一設備上使用時速度很快。

例如:

a = tf.constant([0.78])
a_identity = tf.identity(a)
a.numpy()
array([0.78], dtype=float32)
a_identity.numpy()
array([0.78], dtype=float32)

在變量上調用tf.identity 將生成一個表示該變量在調用時的值的張量。這相當於調用 <variable>.read_value()

a = tf.Variable(5)
a_identity = tf.identity(a)
a.assign_add(1)
<tf.Variable ... shape=() dtype=int32, numpy=6>
a.numpy()
6
a_identity.numpy()
5

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.identity。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。