TensorFlow repeat function which is not available for TF 2.0
tf_repeat(a, dim)
a
tensor
dim
dimension for repeating