/usr/local/lib64/python3.6/site-packages/torch/nn/utils/__pycache__/
/usr/local/lib64/python3.6/site-packages/torch/nn/utils/__pycache__
NameSizeModeActions
clip_grad.cpython-36.pyc39470644editdlrm
convert_parameters.cpython-36.pyc23450644editdlrm
fusion.cpython-36.pyc15280644editdlrm
init.cpython-36.pyc22840644editdlrm
memory_format.cpython-36.pyc37630644editdlrm
parametrizations.cpython-36.pyc149990644editdlrm
parametrize.cpython-36.pyc198700644editdlrm
prune.cpython-36.pyc455570644editdlrm
rnn.cpython-36.pyc159340644editdlrm
spectral_norm.cpython-36.pyc96420644editdlrm
weight_norm.cpython-36.pyc46070644editdlrm
__init__.cpython-36.pyc7320644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/nn/utils/__pycache__/fusion.cpython-36.pyc (1528B)
3 ûEgsã@s4ddlZddlZdd„Zdd„Zdd„Zdd „ZdS) éNcCsN|jp |j stdƒ‚tj|ƒ}t|j|j|j|j|j |j|jƒ\|_|_|S)NzFusion only for eval!) ÚtrainingÚAssertionErrorÚcopyÚdeepcopyÚfuse_conv_bn_weightsÚweightÚbiasÚ running_meanÚ running_varÚeps)ÚconvÚbnZ fused_conv©rúA/usr/local/lib64/python3.6/site-packages/torch/nn/utils/fusion.pyÚfuse_conv_bn_evals    rcCs˜|dkrtj|ƒ}|dkr$tj|ƒ}|dkr6tj|ƒ}tj||ƒ}|||jdgdgt|jƒdƒ}|||||}tjj|ƒtjj|ƒfS)Nééÿÿÿÿ) ÚtorchÚ zeros_likeZ ones_likeÚrsqrtZreshapeÚlenÚshapeÚnnÚ Parameter)Zconv_wZconv_bÚbn_rmÚbn_rvÚbn_epsÚbn_wÚbn_bZ bn_var_rsqrtrrrrs   (rcCsN|jp |j stdƒ‚tj|ƒ}t|j|j|j|j|j |j|jƒ\|_|_|S)NzFusion only for eval!) rrrrÚfuse_linear_bn_weightsrrr r r )Zlinearr Z fused_linearrrrÚfuse_linear_bn_evals   r c CsZ|dkrtj|ƒ}|tj||ƒ}||jdƒ}||||} tjj|ƒtjj| ƒfS)Nrr)rrrZ unsqueezerr) Zlinear_wZlinear_brrrrrZbn_scaleZfused_wZfused_brrrr(s  r)rrrrr rrrrrÚs