topobench.transforms.feature_liftings.set module#

Set lifting for r-cell features to r+1-cell features.

class Set(**kwargs)#

Bases: BaseTransform

Lift r-cell features to r+1-cells by set operations.

Parameters:
**kwargsoptional

Additional arguments for the class.

__init__(**kwargs)#
forward(data)#

Apply the lifting to the input data.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The lifted data.

lift_features(data)#

Concatenate r-cell features to r+1-cell structures.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The lifted data.