B
|
_© ã @ s& d Z ddlmZ G dd dejZdS )z9
|
@author: liaoxingyu
|
@contact: sherlockliao01@gmail.com
|
é )Únnc s& e Zd Zd fdd Zdd Z ZS )ÚSELayeré c sd t t| ¡ t d¡| _t tj|t|| ddtj ddtjt|| |ddt
|
¡ ¡| _ d S )Né F)ÚbiasT)Úinplace) Úsuperr Ú__init__r ÚAdaptiveAvgPool2dÚavg_poolÚ
|
SequentialÚLinearÚintÚReLUÚSigmoidÚfc)ÚselfÚchannelÚ reduction)Ú __class__© ú./layers/se_layer.pyr s
|
zSELayer.__init__c C sF | ¡ \}}}}| |¡ ||¡}| |¡ ||dd¡}|| |¡ S )Nr )Úsizer Úviewr Ú expand_as)r ÚxÚbÚcÚ_Úyr r r Úforward s zSELayer.forward)r )Ú__name__Ú
|
__module__Ú __qualname__r r Ú __classcell__r r )r r r
|
s
|
r N)Ú__doc__Útorchr ÚModuler r r r r Ú<module> s
|
