在张量流中创建张量数组?

时间:2016-10-15 23:36:22

标签: python arrays numpy tensorflow

假设我有两个张量变量,每个变量大小为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数组。我也尝试将其初始化为张量流变量,即。 tf.Variable(tf.zeros([2]))但这也不起作用。

注意,ab是动态输入,即。它们是tf.placeholder变量。

2 个答案:

答案 0 :(得分:2)

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

答案 1 :(得分:0)

作为可能相关的更新,in TensorFlow 1.0 tf.pack()已重命名为tf.stack()