如何将输出层连接到另一个神经网络的输入层？

时间：2020-06-01 19:59:59

标签： python tensorflow deep-learning neural-network reinforcement-learning

Actor网络有5个表示状态值的输入神经元，并将产生一个由一个输出神经元保持的输出值。

Q网络有6个输入神经元：5个代表状态值，1个代表Actor网络的输出。

我将进行梯度下降以分别训练Actor网络，并使Q网络的权重保持恒定。

我的问题是： 如何使用TensorFlow 2.x将Actor-Network的输出层插入Q网络的输入层？

Neural network structure

1 个答案:

答案 0 :(得分：2)

您可以只使用ŧf.keras.Model API：

actor_model = tf.keras.Model(inputs=...,outputs=...)
Q_model = tf.keras.Model(inputs=actor_model.outputs, outputs=...)
相关问题
最新问题