deepmd.pt.model.model.dipole_model

deepmd.pt.model.model.dipole_model#

Attributes#

DPDipoleModel_

Classes#

DipoleModel

A base class to implement common methods for all the Models.

Module Contents#

deepmd.pt.model.model.dipole_model.DPDipoleModel_[source]#
class deepmd.pt.model.model.dipole_model.DipoleModel(*args, **kwargs)[source]#

Bases: deepmd.pt.model.model.dp_model.DPModelCommon, DPDipoleModel_

A base class to implement common methods for all the Models.

model_type = 'dipole'[source]#
translated_output_def()[source]#
forward(coord, atype, box: torch.Tensor | None = None, fparam: torch.Tensor | None = None, aparam: torch.Tensor | None = None, do_atomic_virial: bool = False) → dict[str, torch.Tensor][source]#
forward_lower(extended_coord, extended_atype, nlist, mapping: torch.Tensor | None = None, fparam: torch.Tensor | None = None, aparam: torch.Tensor | None = None, do_atomic_virial: bool = False)[source]#