Skip to main content
 首页 » 编程设计

python之在 tensorflow 中创建张量数组

2024年10月17日31虾米哥

假设我有两个大小为 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() 可能就是您正在寻找的。