deepmd.tf2.utils._dpmodel
=========================

.. py:module:: deepmd.tf2.utils._dpmodel


Functions
---------

.. autoapisummary::

   deepmd.tf2.utils._dpmodel.build_neighbor_list
   deepmd.tf2.utils._dpmodel.extend_coord_with_ghosts
   deepmd.tf2.utils._dpmodel.format_nlist
   deepmd.tf2.utils._dpmodel.inter2phys
   deepmd.tf2.utils._dpmodel.normalize_coord
   deepmd.tf2.utils._dpmodel.to_face_distance


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

.. py:function:: build_neighbor_list(coord: Any, atype: Any, nloc: int, rcut: float, sel: int | list[int], distinguish_types: bool = True) -> deepmd.tf2.env.Array

.. py:function:: extend_coord_with_ghosts(coord: Any, atype: Any, cell: Any | None, rcut: float) -> tuple[deepmd.tf2.env.Array, deepmd.tf2.env.Array, deepmd.tf2.env.Array]

.. py:function:: format_nlist(extended_coord: Any, nlist: Any, nsel: int, rcut: float) -> deepmd.tf2.env.Array

.. py:function:: inter2phys(coord: Any, cell: Any) -> deepmd.tf2.env.Array

.. py:function:: normalize_coord(coord: Any, cell: Any) -> deepmd.tf2.env.Array

.. py:function:: to_face_distance(cell: Any) -> deepmd.tf2.env.Array