6d781d6720bcaeb5aa75874180e5d636e9f4248f,scattering/scattering1d/backend/backend_torch.py,,modulus_complex,#Any#,250

Before Change

    
    // take the stable modulus
    real = modulus(x)
    imag = real.data.new(real.shape).fill_(0.)
    res = torch.cat([real.unsqueeze(-1), imag.unsqueeze(-1)], dim=-1)
    return res

After Change

    // take the stable modulus
    norm = modulus(x)

    res = torch.zeros_like(x)
    res[...,0] = norm

    return res

Instances