mindspore.ops.LJForceWithPMEDirectForce

class mindspore.ops.LJForceWithPMEDirectForce(*args, **kwargs)[source]

Calculate the Lennard-Jones force and PME direct force together.

The calculation formula of Lennard-Jones part is the same as operator LJForce(), and the PME direct part is within PME method.

Agrs:

atom_numbers(int32): the number of atoms, n. cutoff_square(float32): the square value of cutoff. pme_beta(float32): PME beta parameter, same as operator PMEReciprocalForce().

Inputs:
  • uint_crd (Tensor, uint32) - [n, 3], the unsigned int coordinate value of each atom.

  • LJtype (Tensor, int32) - [n,], the Lennard-Jones type of each atom.

  • charge (Tensor, float32) - [n,], the charge carried by each atom.

  • scaler (Tensor, float32) - [3,], the scale factor between real space coordinate and its unsigned int value.

  • nl_numbers - (Tensor, int32) - [n,], the each atom.

  • nl_serial - (Tensor, int32) - [n, 800], the neighbor list of each atom, the max number is 800.

  • d_LJ_A (Tensor, float32) - [q,], the Lennard-Jones A coefficient of each kind of atom pair. q is the number of atom pair.

  • d_LJ_B (Tensor, float32) - [q,], the Lennard-Jones B coefficient of each kind of atom pair. q is the number of atom pair.

Outputs:
  • frc (Tensor, float32), [n, 3], the force felt by each atom.

Supported Platforms:

GPU