def forward(self, input_r, input_i): return complex_dropout2d(input_r, input_i, self.p, self.inplace)
def forward(self,input): if self.training: return complex_dropout2d(input,self.p) else: return input
def forward(self, input_r, input_i): return complex_dropout2d(input_r, input_i, self.p, self.inplace)
def forward(self,input): if self.training: return complex_dropout2d(input,self.p) else: return input