TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.keras.backend.count_params

TensorFlow 1 version View source on GitHub

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

Aliases:

tf.keras.backend.count_params(x)

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)