sponge.function.lennard_jones_potential

查看源文件
sponge.function.lennard_jones_potential(epsilon: Tensor, sigma: Tensor, r_ij: Tensor, mask: Tensor = None)[源代码]

使用 \(\epsilon\)\(\sigma\) 计算Lennard-Jones (LJ) 势。

\[E_{lj}(r_{ij}) = 4 \epsilon \left [\left ( \frac{\sigma}{r_{ij}} \right ) ^{12} - \left ( \frac{\sigma}{r_{ij}} \right ) ^{6} \right]\]
参数:
  • epsilon (Tensor) - 张量的shape为 \((...)\) 。数据类型为float。井深 \(\epsilon\)

  • sigma (Tensor) - 张量的shape为 \((...)\) 。数据类型为float。特征距离 \(\sigma\)

  • r_ij (Tensor) - 张量的shape为 \((...)\) 。数据类型为float。原子 \(i\)\(i\) 之间的距离 \(r_{ij}\)

  • mask (Tensor) - 张量的shape为 \((...)\) 。数据类型为bool。距离的掩码 \(r_{ij}\)。默认值: None

返回:

Tensor。E_coulomb。张量的shape为 \((...)\) 。数据类型为float。

支持平台:

Ascend GPU