Pytorch: как я могу обучать/предсказывать только подмножество узлов в GNN

Question

Я пытаюсь разработать графовую нейронную сеть для проблемы с ограничениями на подмножество узлов. Как я могу навязать, что некоторые узлы включаются в оценки других узлов (передача сообщений), но их собственные значения не меняются (не обучаются и не прогнозируются, поскольку они являются граничными условиями).

Спасибо за вашу помощь!