替代tf.parallel_stack()的即时执行

Question

我想使用tf.parallel_stack()在急于编译的模型中进行数据操作，但是似乎tf.parallel_stack()不支持急于执行。因此，我正在寻找实现相同结果的替代函数。

下面是所需结果的一个小示例:

x = tf.constant([1, 4])
y = tf.constant([2, 5])
z = tf.constant([3, 6])
# Wanted result:
tf.parallel_stack([x, y, z])  # [[1, 4], [2, 5], [3, 6]]

我尝试了tf.concat()，但我仍然需要将得到的1D向量转换成如上所述的二维向量。

tf.concat((x, y, z), axis = 0) # [1, 4, 2, 5, 3, 6] 

Answer 1

也许可以尝试添加一个新的维度:

tf.concat((x[None, ...], y[None, ...], z[None, ...]), axis = 0)