tf.identity

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

tf.identity(
    input, name=None
)

Используется в блокнотах

Возвращаемое значение — это не тот же тензор, что и исходный, но оно содержит те же значения. Эта операция быстрая, когда используется на одном устройстве.

Например:

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

Эта функция также может использоваться для явного переноса тензоров между устройствами. Например, для переноса тензора в памяти GPU обратно в память хоста можно использовать:

with tf.device("/gpu:0"):
  x_on_gpu = tf.constant(1)
with tf.device("/cpu:0"):
  x_on_cpu = tf.identity(x_on_gpu)
x_on_cpu.device
'/job:localhost/replica:0/task:0/device:CPU:0'