View source on GitHub |
Returns the static number of elements in a variable or tensor.
tf.keras.backend.count_params(
x
)
x
: Variable or tensor.Integer, the number of scalars in x
.
>>> 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)