Docs4dev
Docs
tensorflow / 2.3 / keras / backend / count_params.html

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
x Variable or tensor.
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)

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/keras/backend/count_params