rllm.data.BaseGraph¶

class rllm.data.BaseGraph[source]¶

Bases: object

An abstract base class for graph data storage.

clone(*args: str)[source]¶: Performs cloning of tensors for the ones given in *args

cpu(*args: str)[source]¶: Moves the dataset to CPU memory.

cuda(device: int | str | None = None, *args: str, non_blocking: bool = False)[source]¶: Moves the dataset toto CUDA memory.

keys()[source]¶: Returns a list of all graph attribute names.

to(device: int | str, *args: str, non_blocking: bool = False)[source]¶: Performs device conversion of the whole dataset.