nn_models.py 文件源码

def _propagate(self, Y, dropout=0.):
        relu_1 = F.clipped_relu(self.relu_1(Y, dropout=dropout))
        N_mask = F.sigmoid(self.noise_mask_estimate(relu_1))
        X_mask = F.sigmoid(self.speech_mask_estimate(relu_1))
        return N_mask, X_mask