如何将张量dtype = tf.float32_ref转换为dtype = tf.float32?

时间:2019-01-06 08:19:28

标签: tensorflow

我想通过功能float32_ref使用将word_embeddings dtype从float32修改为tf.cast()

   word_embeddings_modify=tf.cast(word_embeddings,dtype=tf.float32)

但是它没有按预期工作,并且word_embeddings_modify dtype仍然为tf.float32_ref。

   word_embeddings = tf.scatter_nd_update(var_output, error_word_f,sum_all)
   word_embeddings_modify=tf.cast(word_embeddings,dtype=tf.float32)
   word_embeddings_dropout = tf.nn.dropout(word_embeddings_2, dropout_pl)

1 个答案:

答案 0 :(得分:1)

您可以使用tf.identity取消引用_ref类型

word_embeddings = tf.identity(word_embeddings)