3 Nba@s ddljZGdddejZdS)Ncs&eZdZdfdd ZddZZS)SELayercs\tt|jtjd|_tjtj|||ddtjddtj|||ddtj |_ dS)NF)biasT)inplace) superr__init__nnAdaptiveAvgPool2davg_pool SequentialLinearReLUSigmoidfc)selfchannel reduction) __class__S/home/ec2-user/SageMaker/lama/saicinpainting/training/modules/squeeze_excitation.pyrs  zSELayer.__init__cCsJ|j\}}}}|j|j||}|j|j||dd}||j|}|S)Nr)sizer viewr expand_as)rxbc_yresrrrforwards zSELayer.forward)r)__name__ __module__ __qualname__rr __classcell__rr)rrrs r)torch.nnr Modulerrrrrs