tf.keras.backend.count_params

TensorFlow 1 version View source on GitHub

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

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)