Tensorflow 堆栈向量出循环以创建矩阵

Question

我有一个 for 循环可以创建等长的向量（tf 张量），比如

a1 = [0, 2, 4 ... ]
a2 = [1, 4, 6 ... ]
...

我想沿着第 0 轴将这些向量连接成一个矩阵

matrix = [[0,2,4...] , [1,4,6...] ... ]

我可以做一个

matrix = tf.concat(0, [matrix, a])

在 for 循环中。然而，第一次迭代不起作用，因为矩阵不存在，如果我将它初始化为一个向量，我就会在结束矩阵的顶部使用该向量。有没有快速的方法来做到这一点？

Answer 1

您可以使用 tf.stack:

matrix = tf.stack([a1, a2, ...])