45bc77266ba893c3ec65cc8661a1b66abe354671,kymatio/scattering3d/tests/test_scattering3d.py,,test_scattering_methods,#,168

Before Change

        Sx = scattering(x, method="integral")

def test_scattering_methods():
    if backend.NAME == "skcuda":
        warnings.warn(("The skcuda backend is not yet implemented for 3D "
            "scattering, but that"s ok (for now)."), RuntimeWarning,
            stacklevel=2)
        return

    shape = (32, 32, 32)
    J = 4
    L = 3
    sigma_0 = 1

After Change

    scattering = Scattering3D(J=J, shape=shape, L=L, sigma_0=sigma_0)

    if not "cpu" in devices:
        x = x.cuda()
        scattering.cuda()

    Sx = scattering(x, method="standard")
    Sx = scattering(x, method="standard", rotation_covariant=False)

    points = torch.zeros(1, 1, 3)

Instances