2

我想制作包含多个输入的 CNN 模型。但是没有任何例子。没有Keras,我可以制作多个输入模型吗?

我的模型有 3 个输入。在第一个输入(atom_in_fea[8,9])通过嵌入(keras 中的线性或密集)层后,输出应与第二个(nbr_fea_idx[8,12])和第三个输入(nbr_fea[8,12,4])合并。

atom_in_fea[8,9] -> embedding layer -> atom_in_fea[8,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)

atom_nbr_fea 形状为 [8,12,4]

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

我建立了一个模型,Keras但有限。如果我能解决这个问题,无论是Keras还是都可以TensorFlow。谢谢

4

0 回答 0