deepmd.pt.model.model.polar_model

Module Contents

Classes

PolarModel

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

Attributes

DPDOSModel_

deepmd.pt.model.model.polar_model.DPDOSModel_

class deepmd.pt.model.model.polar_model.PolarModel(*args, **kwargs)

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

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

model_type = 'polar'

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]

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)