python - 在 TensorFlow 中制作列表并附加到列表中

Question

我是 TensorFlow 的新手。我无法理解如何在 TensorFlow 中创建动态“pythonic”列表。基本上，我对张量对象 (train_data[i]) 执行一些计算并将其附加到“列表”X，我希望它成为形状为 (100,)

我想做这样的事情:

X = []
for i in range(100):
    q = tf.log(train_data[i]) 
    print(q)    #Tensor("Log:0", shape=(), dtype=float32) 
    X.append(q)

我希望 X 是一个形状为 (100,) 的张量，基本上是一个列向量，它是一个张量对象。如果我运行上面的代码，我会得到一个 TensorObjects 的 python 列表。

Answer 1

如果您想将 X 转换为 (100,) 张量，您可以在 for 循环之后添加 X = tf.stack(X):

X = []
for i in range(100):
  q = tf.log(train_data[i]) 
  print(q)    #Tensor("Log:0", shape=(), dtype=float32) 
  X.append(q)
X = tf.stack(X)

这是一个有用的结构，您可能希望 tf.unstack 一些张量，遍历结果列表，然后使用 tf.stack 返回到一个单张量。