ó rÎ\Zc@spddlZddlZdejjfd„ƒYZdejjfd„ƒYZdejjfd„ƒYZdS(iÿÿÿÿNtIdentitycBseZd„ZRS(cCs|S(N((tselftinput((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pytforwards(t__name__t __module__R(((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyRstSegmentConsensuscBs&eZdd„Zd„Zd„ZRS(icCs||_||_d|_dS(N(tconsensus_typetdimtNonetshape(RRR((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyt__init__ s  cCs^|jƒ|_|jdkr<|jd|jdtƒ}n|jdkrT|}nd}|S(NtavgRtkeepdimtidentity(tsizeR RtmeanRtTrueR (Rt input_tensortoutput((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyRs cCsZ|jdkr8|j|jƒt|j|jƒ}n|jdkrP|}nd}|S(NR R(RtexpandR tfloatRR (Rt grad_outputtgrad_in((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pytbackwards ) (RRR RR(((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyR s  tConsensusModulecBseZdd„Zd„ZRS(icCs;tt|ƒjƒ|dkr%|nd|_||_dS(NtrnnR(tsuperRR RR(RRR((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyR )scCst|j|jƒ|ƒS(N(RRR(RR((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyR.s(RRR R(((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyR's ( ttorchtmathtnntModuleRtautogradtFunctionRR(((sP/data/vision/torralba/deepscene/small-projects/TRN-pytorch-pose/ops/basic_ops.pyts