使用tf.concat在TensorFlow中连接输入张量形状:[5,1,1,100]和[5,5,5,100]

时间:2018-12-14 16:25:24

标签: python-3.x tensorflow concatenation tensor tensorflow-datasets

我正在尝试使用tf.concat命令在TensorFlow中连接两个不同维度的张量,即[5,1,1,100]和[5,5,5,100] [NHWC]。我知道执行此操作有一些技巧,但我不知道该技巧。我在Stackoverflow中进行了搜索,并尝试实施现有的想法,但没有得到预期的结果。如果有人能帮助我解决此问题,将很有帮助。

错误

ValueError: Dimension 0 in both shapes must be equal, but are 1 and 5. Shapes are [1,100] and [5,100]. for 'tower_0/cat_layer/concat' (op: 'ConcatV2') with input shapes: [5,1,1,100], [5,5,5,100], [] and with computed input tensors: input[2] = <1>.

0 个答案:

没有答案