
简短的回答是,你可以将张量从转换
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。
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)