Function check_pytorch

python/dgl/homophily.py:20–25

Check if PyTorch is the backend.

()

def check_pytorch():
    """Check if PyTorch is the backend."""
    if HAS_TORCH is False:
        raise ModuleNotFoundError(
            "This function requires PyTorch to be the backend."
        )


def get_long_edges(graph):