假设我有两个大小为 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 变量。
请您参考如下方法:
tf.pack() 可能就是您正在寻找的。
