tf.keras.backend.count_params

View source on GitHub

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

tf.keras.backend.count_params(
    x
)

Arguments:

Returns:

Integer, the number of scalars in x.

Example:

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