deepmd.pd.utils

Submodules

• deepmd.pd.utils.auto_batch_size
• deepmd.pd.utils.dataloader
• deepmd.pd.utils.dataset
• deepmd.pd.utils.decomp
• deepmd.pd.utils.dp_random
• deepmd.pd.utils.env
• deepmd.pd.utils.env_mat_stat
• deepmd.pd.utils.exclude_mask
• deepmd.pd.utils.finetune
• deepmd.pd.utils.learning_rate
• deepmd.pd.utils.multi_task
• deepmd.pd.utils.neighbor_stat
• deepmd.pd.utils.nlist
• deepmd.pd.utils.preprocess
• deepmd.pd.utils.region
• deepmd.pd.utils.serialization
• deepmd.pd.utils.spin
• deepmd.pd.utils.stat
• deepmd.pd.utils.update_sel
• deepmd.pd.utils.utils

Classes

AtomExcludeMask	Computes the type exclusion mask for atoms.
PairExcludeMask	Computes the type exclusion mask for atom pairs.

Package Contents

class deepmd.pd.utils.AtomExcludeMask(ntypes: int, exclude_types: list[int] = [])

Bases: paddle.nn.Layer

Computes the type exclusion mask for atoms.

reinit(ntypes: int, exclude_types: list[int] = []) → None

get_exclude_types() → list[int]

get_type_mask() → paddle.Tensor

forward(atype: paddle.Tensor) → paddle.Tensor

Compute type exclusion mask for atoms.

Parameters:

atype

The extended atom types. shape: nf x natom

Returns:

mask

The type exclusion mask for atoms. shape: nf x natom Element [ff,ii] being 0 if type(ii) is excluded, otherwise being 1.

class deepmd.pd.utils.PairExcludeMask(ntypes: int, exclude_types: list[tuple[int, int]] = [])

Bases: paddle.nn.Layer

Computes the type exclusion mask for atom pairs.

reinit(ntypes: int, exclude_types: list[tuple[int, int]] = []) → None

get_exclude_types() → set[tuple[int, int]]

forward(nlist: paddle.Tensor, atype_ext: paddle.Tensor) → paddle.Tensor

Compute type exclusion mask.

Parameters:

nlist

The neighbor list. shape: nf x nloc x nnei

atype_ext

The extended aotm types. shape: nf x nall

Returns:

mask

The type exclusion mask of shape: nf x nloc x nnei. Element [ff,ii,jj] being 0 if type(ii), type(nlist[ff,ii,jj]) is excluded, otherwise being 1.