a .=icÎã@s6dZddlmZGdd„deƒZGdd„dejƒZdS)z3Functionality for Python <-> C++ frontend inter-op.é)Únnc@s\eZdZdZdd„Zedd„ƒZdd„Zdd „Zd d„Z dd „Z dd„Zdd„Zdd„Z dS)ÚOrderedDictWrapperaª A wrapper around a C++ OrderedDict that dynamically evaluates the OrderedDict getter on a bound C++ module, such that new changes on the C++ side are picked up. Otherwise accessing e.g. ``cpp_module._parameters`` just once would get a frozen copy of the parameters at the time of access. ``torch.nn.Module`` accesses ``_parameters`` et al. via ``self.__dict__`` so using properties does not work. cCs||_||_dS©N)Ú cpp_moduleÚattr©Úselfrr©r ú]/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/nn/cpp.pyÚ__init__szOrderedDictWrapper.__init__cCst|j|jƒSr)Úgetattrrr©rr r r Úcpp_dictszOrderedDictWrapper.cpp_dictcCs |j ¡Sr)rÚitemsr r r r rszOrderedDictWrapper.itemscCs |j ¡Sr)rÚkeysr r r r rszOrderedDictWrapper.keyscCs |j ¡Sr)rÚvaluesr r r r r!szOrderedDictWrapper.valuescCs |j ¡Sr)rÚ__iter__r r r r r$szOrderedDictWrapper.__iter__cCs |j ¡Sr)rÚ__len__r r r r r'szOrderedDictWrapper.__len__cCs|j |¡Sr)rÚ__contains__©rÚkeyr r r r*szOrderedDictWrapper.__contains__cCs|j |¡Sr)rÚ__getitem__rr r r r-szOrderedDictWrapper.__getitem__N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rÚpropertyrrrrrrrrr r r r rs rcsJeZdZdZ‡fdd„Zdd„Zedd„ƒZejdd„ƒZd d „Z ‡Z S)Ú ModuleWrapperzj A subclass of ``torch.nn.Module`` that wraps a C++ frontend module and delegates all access. csh||_tt|ƒ ¡t|dƒ|_t|dƒ|_t|dƒ|_t|ƒD]"}| d¡s@t ||t|j|ƒƒq@dS)NÚ_parametersÚ_buffersÚ_modulesÚ_)rÚsuperrrrrrr ÚdirÚ startswithÚsetattrrr©Ú __class__r r r7s zModuleWrapper.__init__cCsR| ¡D]*}||jƒ|_|jdur||jjƒ|j_q| ¡D]}||jƒ|_q<|Sr)Ú parametersÚdataZ_gradÚbuffers)rÚfnÚparamÚbufr r r Ú_applyDs zModuleWrapper._applycCs|jjSr)rÚtrainingr r r r r/RszModuleWrapper.trainingcCs|j |¡dSr)rÚtrain)rÚmoder r r r/VscCs |j ¡Sr)rÚ__repr__r r r r r2ZszModuleWrapper.__repr__)rrrrrr.rr/Úsetterr2Ú __classcell__r r r&r r1s rN)rZtorchrÚobjectrÚModulerr r r r Ús+