a 8Sicz@sddlZddlmZmZmZmZmZmZddlZddlm Z dgZ ddddZ dd Z e e dd d d Zejd ee e feddddZedd ee e edddZdd ee e feee efdddZdS)N)AnyCallableDictIteratorListTuple)Tensorfunctional_call)returncsj|j|jttdfdd }ttddfdd }tdjf||d}||_|_dS) N)namer cs |vr|S||SN)__getattribute__)selfr Z attr_to_pathclsparams_and_buffersT/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/nn/utils/stateless.py _getattributes z$_change_class.._getattribute)r valuer cs(|vr||<n|||SdSr ) __setattr__)rr rrrr_setattrsz_change_class.._setattrZStatelessReplacer)r r) __class__ _attr_to_pathstrrtype__name__ _orig_class)modulerrr param_clsrrr _change_class s  r cstttddfdd }|S)N) tensor_name full_pathtensorr cs4t|dr||j|<ni|_||j|<t|dS)Nr)hasattrrr )rr!r"r#rrr_swap_parameters's    z-_create_swap_params.._swap_parameters)rr)rr&rr%r_create_swap_params&s r')r r"r cCs*t|dr&|j|_t|dt|ddS)Nrr)r$rrdelattr)rr r"rrr _remove_swap3s  r)ztorch.nn.Module)rparameters_and_buffersr c cs|D]$\}}tt|||d||fqz*dVW|D]}tt||d|dq>> a = {'foo': torch.zeros(())} >>> mod = Foo() # does self.foo = self.foo + 1 >>> print(mod.foo) # tensor(0.) >>> functional_call(mod, a, torch.ones(())) >>> print(mod.foo) # tensor(0.) >>> print(a['foo']) # tensor(1.) Args: module (torch.nn.Module): the module to call parameters_and_buffers (dict of str and Tensor): the parameters that will be used in the module call. args (tuple): arguments to be passed to the module call kwargs (dict): keyword arguments to be passed to the module call Returns: Any: the result of calling ``module``. z3The stateless API can't be used with Jitted modulesN) torchjit is_tracing is_scripting isinstanceRecursiveScriptModule ScriptModuleScriptFunction RuntimeErrorr/tuple)rr*r3r7outrrrr Ys$'   .)N) contextlibtypingrrrrrrr8r__all__r r'rr)contextmanagerr/r-r rrrrs2