TPD
===

.. py:module:: topmost.models.hierarchical.TraCo.TPD



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



.. autoapisummary::


   topmost.models.hierarchical.TraCo.TPD.TPD





.. py:class:: TPD(sinkhorn_alpha, sinkhorn_max_iter=5000, stopThr=0.005)

   Bases: :py:obj:`torch.nn.Module`


   .. py:attribute:: sinkhorn_alpha


   .. py:attribute:: sinkhorn_max_iter
      :value: 5000



   .. py:attribute:: stopThr
      :value: 0.005



   .. py:attribute:: epsilon
      :value: 1e-16



   .. py:method:: forward(topic_embeddings_list, weight_loss_TPD=20.0)


   .. py:method:: sinkhorn(M, return_transp=False)