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

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)