5c6c50274c11ecbf9be54052394922fabde80923,kymatio/scattering2d/core/scattering2d.py,,scattering2d,#,4

Before Change

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

    out_S = concatenate(out_S)
    return out_S

After 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

Instances