tf.size

Returns the size of a tensor.