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

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

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)

以上内容是否对您有帮助:
二维码
建议反馈
二维码