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'