Exemplo n.º 1
0
 def forward(self, input_r, input_i):
     return complex_dropout(input_r, input_i, self.p, self.inplace)
Exemplo n.º 2
0
 def forward(self,input):
     if self.training:
         return complex_dropout(input,self.p)
     else:
         return input