nn_models.py 文件源码

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