tf.repeat

import tensorflow as tf
tf.repeat(['a', 'b', 'c'], repeats=[3, 0, 2], axis=0)

<tf.Tensor: shape=(5,), dtype=string, numpy=array([b'a', b'a', b'a', b'c', b'c'], dtype=object)>

repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=0)

<tf.Tensor: shape=(5, 2), dtype=int32, numpy=
array([[1, 2],
       [1, 2],
       [3, 4],
       [3, 4],
       [3, 4]], dtype=int32)>

repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=1)

<tf.Tensor: shape=(2, 5), dtype=int32, numpy=
array([[1, 1, 2, 2, 2],
       [3, 3, 4, 4, 4]], dtype=int32)>

 repeat([[1,2], [3,4]], repeats=2)

 <tf.Tensor: shape=(8,), dtype=int32, numpy=array([1, 1, 2, 2, 3, 3, 4, 4], dtype=int32)>