阅读(5.9k) 书签 (0)

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

2019-03-22 18:25 更新

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:如果值无效。