最近写代码处理一个需求
任务:复制张量元素个数
找到了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(),是按照维度复制张量。