a852f08fcf6679826771992fcf162c3691dff187,kymatio/scattering2d/tests/test_scattering2d.py,,,#,11

Before Change

from kymatio.scattering2d import backend


if torch.cuda.is_available():
    devices = ["cuda", "cpu"]
else:
    devices = ["cpu"]


def reorder_coefficients_from_interleaved(J, L):
    // helper function to obtain positions of order0, order1, order2 from interleaved
    order0, order1, order2 = [], [], []
    n_order0, n_order1, n_order2 = 1, J * L, L ** 2 * J * (J - 1) // 2

After Change

devices = ["cpu"]
if torch.cuda.is_available():
    devices.append("cuda")

def reorder_coefficients_from_interleaved(J, L):
    // helper function to obtain positions of order0, order1, order2 from interleaved
    order0, order1, order2 = [], [], []
    n_order0, n_order1, n_order2 = 1, J * L, L ** 2 * J * (J - 1) // 2

Instances