gnn_tracking.models.dynamic_edge_conv

Module Contents

Classes

DynamicEdgeConv

Attributes

knn

gnn_tracking.models.dynamic_edge_conv.knn

class gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv(nn: Callable, k: int, aggr: str = 'max', num_workers: int = 1, **kwargs)

Bases: torch_geometric.nn.conv.MessagePassing

reset_parameters()

get_edge_index()

forward(x: torch.Tensor | torch_geometric.typing.PairTensor, batch: torch_geometric.typing.OptTensor | torch_geometric.typing.PairTensor | None = None) → torch.Tensor

message(x_i: torch.Tensor, x_j: torch.Tensor) → torch.Tensor

__repr__() → str