TensorFlow函数教程:tf.keras.backend.set_floatx

由 Carrie 创建, 最后一次修改 2019-03-22

tf.keras.backend.set_floatx函数

tf.keras.backend.set_floatx(value)

定义在:tensorflow/python/keras/backend.py。

设置默认的浮点类型。

参数:

  • value:字符串;'float16','float32'或'float64'。

例:

>>> from keras import backend as K
>>> K.floatx()
'float32'
>>> K.set_floatx('float16')
>>> K.floatx()
'float16'

可能引发的异常:

  • ValueError:如果值无效。


以上内容是否对您有帮助:

二维码
建议反馈
二维码