tensorflow1.版本和tensorflow2.版本复制张量元素个数

最近写代码处理一个需求

任务：复制张量元素个数

找到了tf.repeat()函数

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

使用时会发现

import tensorflow as tf

sess = tf.Session()
a = tf.constant([[1,2,3],[1,2,3],[1,2,3],[1,2,3]])
print(a)
print(sess.run(a))

b = tf.repeat(a, 2)
print(b)

原因：我的tensorflow版本是1.13的， tf.repeat()是tensorflow2.0以后引入的。

这个时候不知所措…

自从问了bing以后

找到了替代函数 tf.keras.backend.repeat_elements( x, rep, axis )

import tensorflow as tf

sess = tf.Session()
a = tf.constant([[1.1,2,3],[1,2.,3],[1,2,31],[1.11,2,3]])
print(a)
print(sess.run(a))

b = tf.keras.backend.repeat_elements(a, 4, 1)
print(b)
print(sess.run(b))

总结：

tensorflow1. 用tf.keras.backend.repeat_elements()
tensorflow2. 用tf.repeat()

拓展：上述是按元素复制个数。还有一个函数是tf.tile(),是按照维度复制张量。