Tensorflow堆栈向量在循环之外创建矩阵

时间:2017-07-27 11:45:31

标签: tensorflow stack tensor

我有一个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循环中

。然而,第一次迭代不起作用,因为矩阵不存在,如果我将其初始化为向量,我会在末端矩阵的顶部停留该向量。有这么快的方法吗?

1 个答案:

答案 0 :(得分:1)

您可以使用tf.stack

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