如何将昏暗N的张量列表转换为暗淡的N + 1张量

时间:2016-03-01 18:05:26

标签: tensorflow

我需要将维数N的张量列表转换为维数为N + 1的新张量,以便新维度成为最正确的维度。

例如,如果x和y是形状(4,3)的张量,那么我试图通过形成z并将张量x设置为第0个元素来创建形状(4,3,2)的新张量z沿第三维度设置张量y作为第三维度的第一个元素。在伪代码中:

z = tf.fromList([x,y],3)

Tensorflow中最好的方法是什么?我无法从TF 0.7.1的文档中找到它。

2 个答案:

答案 0 :(得分:12)

如果我正确读你,你想交错两个张量的数据。

您希望tf.pack()将它们组合在一起,这将形成一个形状张量[2, 4, 3],然后tf.transpose([1, 2, 0]),从而产生张量以达到您想要的交错。

答案 1 :(得分:10)

dga的方法有效,但{T}已从TensorFlow V1.0开始删除tf.pack()

您可以使用tf.stack()来实现相同目标。

文档:https://www.tensorflow.org/api_docs/python/tf/stack