Tensorflow循环外叠加向量以创建矩阵

我有一个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循环内部。但是第一次迭代不起作用，因为矩阵不存在，如果我将它初始化为一个向量，我将被困在结束矩阵的顶部。有没有快速的方法

您可以使用：

---

你的意思是将a1、a2附加到一个空列表中，并在列表上调用堆栈？是的，这可能是一种方法。

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