簡體   English   中英

如何將 output 層連接到另一個神經網絡的輸入層?

[英]How to connect output layers to the input layer of another neural network?

Actor-network 有 5 個輸入神經元代表 state 值,並將產生一個 output 值,由一個 output 神經元持有。

Q Network 有 6 個輸入神經元:5 個代表 state 值,1 個代表 Actor-network 的 output。

我將進行梯度下降來單獨訓練 Actor 網絡,保持 Q 網絡的權重不變。

我的問題是:我應該如何構造將 Actor-Network 的 output 層插入 Q 網絡的輸入層,TensorFlow 2.x?

神經網絡結構

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

actor_model = tf.keras.Model(inputs=...,outputs=...)
Q_model = tf.keras.Model(inputs=actor_model.outputs, outputs=...)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM