32d73ccb0d80db3727775fb9928f5f151ddaaab5,nt/speech_enhancement/mask_estimation.py,,simple_ideal_soft_mask,#,44

Before Change

        X = input[0]

    // Permute if nessesary
    if feature_dim != -2 or source_dim != -1:
        r = list(range(np.ndim(X)))
        r[feature_dim], r[-2] = r[-2], r[feature_dim]
        r[source_dim], r[-1] = r[-1], r[source_dim]
        X = np.transpose(X, axes=r)

    power = np.einsum("...dk,...dk->...k", X.conjugate(), X)
    mask = (power / np.sum(power, axis=power.ndim-1, keepdims=True)).real

    return mask

After Change

    if not tuple_output:
        return np.squeeze(mask, axis=feature_dim)
    else:
        sizes = np.cumsum([o.shape[source_dim] for o in input])
        output = np.split(mask, sizes[:-1], axis=source_dim)

        for i in range(len(output)):
            if output[i].shape[source_dim] is 1:

Instances