tglite.TGraph

class tglite.TGraph(edges: ndarray, times: ndarray, num_nodes: int = None)

A container for temporal graph and related tensor data. It initially stores temporal edges in COO format, sorted based on timestamp. While performing neighborhood sampling, it uses CSR format for faster lookups. TGLite automatically handles the construction and management of these graph formats without intervention from the user.

__init__(edges: ndarray, times: ndarray, num_nodes: int = None)

Internal constructor for creating a TGraph

Parameters:

• edges (np.ndarray) –

• times (np.ndarray) –

• num_nodes (int) –

Query graph structure

num_nodes()	Total number of nodes
num_edges()	Total number of edges

Query graph data

efeat	Returns edge feature
nfeat	Returns node feature
edata	Returns edge data
ndata	Returns node data

Query memory-based TGN data

mem	Returns node memory
mailbox	Returns node mailbox

Set device

storage_device()	Returns TGraph's storage device
compute_device()	Returns TGraph's computing device
set_compute(device)	Sets computing device
move_data(device, **kwargs)	Moves tensor data to device while keeping graph on CPU