sponge.function.coulomb_interaction

sponge.function.coulomb_interaction(q_i: Tensor, q_j: Tensor, r_ij: Tensor, mask: Tensor = None, coulomb_const: float = 1)

计算库仑相互作用。

$$E_{coulomb}(r_{ij})=k\frac{q_i{q}_j}{r_{ij}}$$

参数：

• q_i (Tensor) - 张量的shape为 $(...)$。数据类型为float。原子 $q_i$-th 。

• q_j (Tensor) - 张量的shape为 $(...)$。数据类型为float。原子 $q_j$-th 。

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

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

• coulomb_const (float) - 库仑常量 $k$ 。默认值：1

返回：

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

支持平台：

Ascend GPU