o %ݫi@sBdZddlZddlmZddlmZeeZGdddejZ dS)z?Library implementing dropout. Authors * Mirco Ravanelli 2020 N) get_loggercs*eZdZdZdfdd ZddZZS) Dropout2daThis function implements dropout 2d. It randomly put zeros on entire channels. Arguments --------- drop_rate : float It is the dropout factor (between 0 and 1). inplace : bool If True, it uses inplace operations. Example ------- >>> drop = Dropout2d(drop_rate=0.5) >>> inputs = torch.rand(10, 50, 40) >>> output=drop(inputs) >>> output.shape torch.Size([10, 50, 40]) Fcs.t||_||_tj|j|jd|_dS)N)pinplace)super__init__ drop_raternnrdrop)selfrr __class__L/home/ubuntu/.local/lib/python3.10/site-packages/speechbrain/nnet/dropout.pyr#s zDropout2d.__init__cCs6|dddd}||}|dddd}|S)a9Applies dropout 2d to the input tensor. Arguments --------- x : torch.Tensor (batch, time, channel1, channel2) input to normalize. 4d tensors are expected. Returns ------- x_drop : torch.Tensor The tensor with channels zeroed out. ) transposer )r xx_droprrrforward)s zDropout2d.forward)F)__name__ __module__ __qualname____doc__rr __classcell__rrr rrsr) rtorchtorch.nnr speechbrain.utils.loggerrrloggerModulerrrrrs