TensorFlow:将float64张量转换为float32

TensorFlow:将float64张量转换为float32,第1张

TensorFlow:将float64张量转换为float32

简短的回答是,你可以将张量从转换

tf.float64
tf.float32
使用
tf.cast()
OP:

loss = tf.cast(loss, tf.float32)

更长的答案是,这不能解决优化器的所有问题。(缺少对支持

tf.float64
的已知问题。)优化器要求
tf.Variable
您要优化的所有对象也必须具有type
tf.float32



欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/zaji/5642867.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-12-16
下一篇2022-12-16

发表评论

登录后才能评论

评论列表(0条)

    保存