Можете ли вы обновить параметры модуля из nn.compact этого модуля? (самоизменяющиеся сети)
Я новичок в льне, и мне было интересно, как правильно получить такое поведение:
param = f.init(key,x)
new_param, y = f.apply(param,x)
Где f — экземпляр nn.module.
Где f может пройти через несколько операций, чтобы получить new_param, и что эти операции могут полагаться на промежуточный параметр для получения своего вывода.
Итак, в принципе, есть ли способ получить доступ и обновить параметры, предоставленные экземпляру nn.module, из__call__
, не теряя при этом функционального свойства, поэтому все это можно обернуть преобразованием функции градиента.