tensorflow - 如何在 tensorflow 中复制变量

Question

在 numpy 中，我可以使用 numpy.copy 创建变量的副本.有没有类似的方法可以用来在 TensorFlow 中创建张量的副本？

Answer 1

在 TF2 中:tf.identity()会为你做好事。最近在google colab中使用该功能遇到了一些问题。如果这就是你在这里的原因，这将对你有所帮助。

Error : Failed copying input tensor from /job:localhost/replica:0/task:0/device:CPU:0 to /job:localhost/replica:0/task:0/device:GPU:0 in order to run Identity: No unary variant device copy function found for direction: 1 and Variant type_index: tensorflow::data::(anonymous namespace)::DatasetVariantWrapper [Op:Identity]

#Erroneous code
tensor1 = tf.data.Dataset.from_tensor_slices([[[1], [2]], [[3], [4]]])
tensor2 = tf.identity(tensor1)

#Correction
tensor1 = tf.data.Dataset.from_tensor_slices([[[1], [2]], [[3], [4]]])
with tf.device('CPU'): tensor2 = tf.identity(tensor1)