a ==icyã@s.ddlmZd dd„Zd dd „Zdd d „ZdS)é)ÚcoreéNcKsx|r|n dddif}|jrH|j |d¡g|dfd|gi|d¤Ž}nt |d|j¡}| |¡|j ||g|g¡S)ÚPReluZ ConstantFillÚvaluegÐ?rZ_slopeÚshaper)Z init_paramsZparam_init_netÚ __getattr__rZScopedBlobReferenceZ AddParameterÚnetr)ÚmodelÚblob_inÚblob_outZ num_channelsZ slope_initÚkwargsZslope©r ús/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/helpers/nonlinearity.pyÚprelu s ÿþýü ÿ rFÚNCHWcKs&|r d|d<|jj||fd|i|¤ŽS)zRelu.ÚCUDNNÚengineÚorder)rZRelu©r r r Z use_cudnnrr r r rÚrelu srcKs&|r d|d<|jj||fd|i|¤ŽS)zTanh.rrr)rZTanhrr r rÚtanh'sr)rN)Fr)Fr)Z caffe2.pythonrrrrr r r rÚs