简短的回答是,你可以将张量从转换
tf.float64到
tf.float32使用
tf.cast()OP:
loss = tf.cast(loss, tf.float32)
更长的答案是,这不能解决优化器的所有问题。(缺少对支持
tf.float64的已知问题。)优化器要求
tf.Variable您要优化的所有对象也必须具有type
tf.float32。

简短的回答是,你可以将张量从转换
tf.float64到
tf.float32使用
tf.cast()OP:
loss = tf.cast(loss, tf.float32)
更长的答案是,这不能解决优化器的所有问题。(缺少对支持
tf.float64的已知问题。)优化器要求
tf.Variable您要优化的所有对象也必须具有type
tf.float32。