DENSEGraph

class marius.data.DENSEGraph

__init__(self: marius._data.DENSEGraph) → None

__init__(self: marius._data.DENSEGraph, hop_offsets: torch.Tensor, node_ids: torch.Tensor, in_offsets: torch.Tensor, in_neighbors_vec: List[torch.Tensor], in_neighbors: torch.Tensor, out_offsets: torch.Tensor, out_neighbors_vec: List[torch.Tensor], out_neighbors: torch.Tensor, num_nodes_in_memory: int) → None

getNeighborIDs(self: marius._data.DENSEGraph, incoming: bool = True, global_ids: bool = False) → torch.Tensor

setNodeProperties(self: marius._data.DENSEGraph, node_properties: torch.Tensor) → None

to(self: marius._data.DENSEGraph, device: torch.device) → None

clear(self: marius._data.DENSEGraph) → None

getLayerOffset(self: marius._data.DENSEGraph) → int

property hop_offsets

property in_neighbors

description of the variable

property in_neighbors_vec

property node_properties

property num_nodes_in_memory

property out_neighbors

property out_neighbors_vec

performMap(self: marius._data.DENSEGraph) → None

prepareForNextLayer(self: marius._data.DENSEGraph) → None