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