tf.repeat(), Tensorflow2.1.0以上

2020.09.18补充
今天处理一个数据用到了np.repeat()
备注一下，这个只能处理二维数组
比如我想对a = [1,2,3]使用，让它竖着重复几次

处理不了一维，会报错
np.repeat(a.T, repeats=k次, axis=1)

要先expand_dims变成二维的才行
np.repeat(np.expand_dims(a, axis=0).T, repeats=k次, axis=1)

今天看代码用到了tf.repeat()

查了下资料没有人专门讲解这个，我来写一下，希望能帮到后人。

---

官方文档：

https://www.tensorflow.org/api_docs/python/tf/repeat?hl=ca

---

调用该方法：

tf.repeat(input, repeats, axis=None, name=None)

参数：
1）input: 一个tensor

2）repeats: 重复的次数
注意：len(repeats) must equal input.shape[axis] if axis is not None

3）axis：维度
这个看例子来理解
说明：An int. The axis along which to repeat values. By default (axis=None), use the flattened input array, and return a flat output array.
如果axis没有参数，则会先flatten数组，变成一维