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。
- 支持平台：
AscendGPU