tf.keras.backend.count_params

tf.keras.backend.count_params(x)

Defined in tensorflow/python/keras/_impl/keras/backend.py.

Returns the static number of elements in a variable or tensor.

Arguments:

• x: Variable or tensor.

Returns:

Integer, the number of scalars in `x`.

Example:

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