如何将一个尺寸相同但另一个尺寸不同的两个张量流张量连接起来?

时间:2020-10-31 08:42:49

标签: tensorflow concatenation one-hot-encoding

我正在尝试使用tensorflow API进行一次热编码。为此,您需要预先指定不同值的数量,因此我不得不遍历每个变量并计算每种情况下的不同值。这为我想要重新结合在一起的每个变量留了一个单幅热编码张量。每个维度在一个维度上具有相同的大小,而在另一个维度上具有不同的大小(取决于不同值的数量)。现在,我想将它们沿着具有匹配大小的轴重新组合成一个。但是,tensorflow concat方法似乎要求它们在两个维度上都必须匹配大小。我是否必须还原到pandas / numpy才能实现此目的?感觉应该很简单。

Tensor input to the concatenation method

Tensorflow error

1 个答案:

答案 0 :(得分:1)

轴应为1而不是0:

import tensorflow as tf
x = tf.random.uniform([100, 100])
y = tf.random.uniform([100, 2])
z = tf.concat((x, y), 1)