首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何将张量dtype=tf.float32_ref转换为dtype=tf.float32?

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

Stack Overflow用户
提问于 2019-01-06 16:19:28
回答 1查看 1K关注 0票数 3

我想通过函数tf.cast()将word_embeddings数据类型从float32_ref修改为float32

代码语言:javascript
复制
   word_embeddings_modify=tf.cast(word_embeddings,dtype=tf.float32)

但它并没有像预期的那样工作,word_embeddings_modify dtype仍然是tf.float32_ref。

代码语言:javascript
复制
   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)
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2019-01-06 20:37:06

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

代码语言:javascript
复制
word_embeddings = tf.identity(word_embeddings)
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/54059805

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档