tf.size

View source on GitHub

tf.size(
    input, out_type=tf.dtypes.int32, name=None
)