gnn_tracking.models.dynamic_edge_conv
=====================================

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


Attributes
----------

.. autoapisummary::

   gnn_tracking.models.dynamic_edge_conv.knn


Classes
-------

.. autoapisummary::

   gnn_tracking.models.dynamic_edge_conv.DynamicEdgeConv


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

.. py:data:: knn
   :value: None


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

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


   .. py:attribute:: nn


   .. py:attribute:: k


   .. py:attribute:: num_workers
      :value: 1



   .. py:attribute:: edge_index
      :value: None



   .. 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