deepmd.pt.model.model.polar_model

deepmd.pt.model.model.polar_model#

Attributes#

DPPolarModel_

Classes#

PolarModel

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

Module Contents#

deepmd.pt.model.model.polar_model.DPPolarModel_[source]#
class deepmd.pt.model.model.polar_model.PolarModel(*args, **kwargs)[source]#

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

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

model_type = 'polar'[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]#