b37fdf43bd950c4872d8ea39975d98a1b5d75866,kymatio/scattering2d/backend/backend_torch.py,,cdgmm,#,161

Before Change

    if B.ndimension() != 3:
        raise RuntimeError("The filters must be simply a complex array!")

    if type(A) is not type(B):
        raise RuntimeError("A and B should be same type!")

After Change

        if inplace:
            return A.mul_(B)
        else:
            return A * B
    else:
        C = A.new(A.size())

        A_r = A[..., 0].contiguous().view(-1, A.size(-2)*A.size(-3))

Instances