Как указать путь расчета градиента в нейронной сети в Pytorch ⇐ Python

[Anonymous]

Я хочу внедрить нейронную сеть на Pytorch, где градиенты не рассчитываются по всем весам. Допустим, например, у меня есть MLP с тремя уровнями, и я хочу, чтобы половина узлов в последнем слое вычислила их обратное распространение вплоть до первого уровня, но другая половина последнего слоя вычисляет только до до тех пор средний слой. Я был бы благодарен за любую помощь. Спасибо

Подробнее здесь: https://stackoverflow.com/questions/794 ... in-pytorch