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

2019-03-10 13:20 更新

tf.keras.backend.count_params函数

tf.keras.backend.count_params(x)

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

返回变量或张量中的静态元素数。

参数:

  • x:变量或张量。

返回:

整数,“x”中的标量数。

示例: python >>> kvar = K.zeros((2,3)) >>> K.count_params(kvar) 6 >>> K.eval(kvar) array([[ 0., 0., 0.], [ 0., 0., 0.]], dtype=float32)

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

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号