LabelBatch

Module to build Graph objects and perform operations on them.

class LabelBatch(*args: Any, **kwargs: Any)

Bases: Batch

Extends the Batch class to include LabelTensor objects.

Example:

>>> import torch
>>> from pina.graph import Graph, LabelBatch
>>> graphs = [Graph(pos=torch.randn(3, 2),
...     edge_index=torch.tensor([[0, 1], [1, 0]])) for _ in range(2)]
>>> batch = LabelBatch.from_data_list(graphs)
>>> batch.num_graphs
2

classmethod from_data_list(data_list)

Create a Batch object from a list of Data or Graph objects.

Parameters:

data_list (list[Data] | list[Graph]) – List of Data or Graph objects.

Returns:

A Batch object containing the input data.

Return type:

Batch