假设我有两个张量变量,每个变量大小为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]))
但这也不起作用。
注意,a
和b
是动态输入,即。它们是tf.placeholder
变量。
答案 0 :(得分:2)
tf.pack()可能就是你要找的东西。
答案 1 :(得分:0)
作为可能相关的更新,in TensorFlow 1.0 tf.pack()
已重命名为tf.stack()
。