tf.identity

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

tf.identity(
    input, name=None
)

Например:

import tensorflow as tf
val0 = tf.ones((1,), dtype=tf.float32)
a = tf.atan2(val0, val0)
a_identity = tf.identity(a)
print(a.numpy())          #[0.7853982]
print(a_identity.numpy()) #[0.7853982]