014963415d1cbc895f8c6f9e7f161950e37dc42e,kymatio/scattering2d/backend/torch_skcuda_backend.py,,_get_dtype,#,20

Before Change

def _get_dtype(t):
    if isinstance(t, torch.cuda.FloatTensor):
        return "float"
    elif isinstance(t, torch.cuda.DoubleTensor):
        return "double"

def _iscomplex(x):

After Change

Stream = namedtuple("Stream", ["ptr"])

def _get_dtype(t):
    dtypes = {torch.float32: "float",
              torch.float64: "double"}

    return dtypes[t.dtype]

def _iscomplex(x):
    return x.shape[-1] == 2

Instances