tf.compat.v2.size

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