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

2019-03-13 13:51 更新

tf.keras.backend.dtype函数

tf.keras.backend.dtype(x)

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

以字符串形式返回Keras张量或变量的dtype。

参数:

  • x:张量或变量。

返回:

字符串,“x”的dtype。

例子:

python >>> from keras import backend as K >>> K.dtype(K.placeholder(shape=(2,4,5))) 'float32' >>> K.dtype(K.placeholder(shape=(2,4,5), dtype='float32')) 'float32' >>> K.dtype(K.placeholder(shape=(2,4,5), dtype='float64')) 'float64' # Keras variable >>> kvar = K.variable(np.array([[1, 2], [3, 4]])) >>> K.dtype(kvar) 'float32_ref' >>> kvar = K.variable(np.array([[1, 2], [3, 4]]), dtype='float32') >>> K.dtype(kvar) 'float32_ref'

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号