예제 #1
0
 def forward(self, input_r, input_i):
     return complex_dropout(input_r, input_i, self.p, self.inplace)
예제 #2
0
 def forward(self,input):
     if self.training:
         return complex_dropout(input,self.p)
     else:
         return input