60c2e01bfc93bc01da21b79ebe9f222e601e1931,kymatio/scattering2d/core/scattering2d.py,,scattering2d,#,4

Before Change

            S_2_r = fft(S_2_c, "C2R", inverse=True)
            S_2_r = unpad(S_2_r)

            S[(n1, n2)] = S_2_r

    if vectorize:
        sorted_keys = sorted(S.keys(), key=lambda x: (len(x),) + x)

        S = [S[key] for key in sorted_keys]

        S = concatenate(S)

    return S

After Change

            out_S_2.append(S_2_r)

    out_S = []
    out_S.extend(out_S_0)
    out_S.extend(out_S_1)
    out_S.extend(out_S_2)

    if vectorize:
        out_S = concatenate(out_S)

    return out_S


__all__ = ["scattering2d"]

Instances