我想制作包含多个输入的 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
。谢谢