在 tensorflow 中創建一個張量數組？

Question

假設我有兩個大小為 2x4 的張量變量：

v1 = tf.logical_and(a1, b1)
v2 = tf.logical_and(a2, b2)

相反，我想將它們存儲在一個名為v的數組中，該數組的大小為 2x2x4。 我如何在 Tensorflow 中做到這一點？ 這個想法是這樣的：

for i in range(2):
  v[i] = tf.logical_and(a[i],b[i])

我如何初始化v ？ 我嘗試將v初始化為一個無效的 numpy 數組。 我還嘗試將其初始化為 tensorflow 變量，即。 tf.Variable(tf.zeros([2]))但這也不起作用。

注意， a和b是動態輸入，即。 它們是tf.placeholder變量。

Answer 1

tf.pack() 可能就是你要找的。

Answer 2

作為一個潛在的相關更新，在 TensorFlow 1.0 中tf.pack()被重命名為tf.stack() 。