Python 如何将多个输入放入tensorflow 2.0中？（不含keras）

我想制作包含多个输入的CNN模型。但这方面没有任何例子。如果没有

Keras

，我可以制作多输入模型吗

我的模型有3个输入。第一个输入（

fea[8,9]

）通过嵌入（keras中的线性或密集）层后，输出应与第二个（

nbr\U fea\u idx[8,12]

）和第三个输入（

nbr\U fea[8,12,4]

）合并

[]是他们的形状

这是我的密码

python, tensorflow

atom_nbr_fea = atom_in_fea[nbr_fea_idx, :] 

final = tf.concat([atom_in_fea.unsqueeze(1).expand(8, 12, 4),atom_nbr_fea, nbr_fea], 2)

原子nbr fea形状为[8,12,4]

最终形状为[8,12,12]

我用

Keras

构建了一个模型，但有限。 如果我能解决这个问题，无论是

Keras

还是

TensorFlow

，都可以。多谢各位

python, tensorflow

atom_nbr_fea = atom_in_fea[nbr_fea_idx, :] 

final = tf.concat([atom_in_fea.unsqueeze(1).expand(8, 12, 4),atom_nbr_fea, nbr_fea], 2)