如何按行重塑张量流张量?

时间:2019-11-29 19:43:01

标签: tensorflow

我有一个张量流张量-

a = [0,0,0,0,1,1]

如何重复a来获得-

b = [0,0,0,0,0,0,0,0,1,1,1,1] # 8 times 0 and 2 times 1

我尝试通过逐行平铺张量并重塑它来实现此目的-

X = tf.concat([[a],[a]], axis=0)
X = tf.reshape(X, [-1,2])
X = tf.reshape(X, [-1]) # assuming that reshape repeats each row

我尝试了上面的几种变体,但是我以如下张量结束

b = [0,0,0,0,1,1,0,0,0,0,1,1]

我缺少对tf.reshape的理解。

1 个答案:

答案 0 :(得分:0)

一个同事刚刚解决了我的问题-

tf.reshape(tf.tile(a[:,None], [1,2]), [-1])

根据此处的行为,看来tf.tiletf.concat -> tf.reshape在内存中的元素排列不同。