:py:mod:`gnn_tracking.models.dynamic_edge_conv`
===============================================

.. py:module:: gnn_tracking.models.dynamic_edge_conv


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv




Attributes
~~~~~~~~~~

.. autoapisummary::

   gnn_tracking.models.dynamic_edge_conv.knn


.. py:data:: knn

   

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


   Bases: :py:obj:`torch_geometric.nn.conv.MessagePassing`

   .. py:method:: reset_parameters()


   .. py:method:: get_edge_index()


   .. py:method:: forward(x: torch.Tensor | torch_geometric.typing.PairTensor, batch: torch_geometric.typing.OptTensor | torch_geometric.typing.PairTensor | None = None) -> torch.Tensor


   .. py:method:: message(x_i: torch.Tensor, x_j: torch.Tensor) -> torch.Tensor


   .. py:method:: __repr__() -> str