sponge.function.coulomb_interaction
- sponge.function.coulomb_interaction(q_i: Tensor, q_j: Tensor, r_ij: Tensor, mask: Tensor = None, coulomb_const: float = 1)[源代码]
计算库仑相互作用。
- 参数:
q_i (Tensor) - 张量的shape为
。数据类型为float。原子 -th 。q_j (Tensor) - 张量的shape为
。数据类型为float。原子 -th 。r_ij (Tensor) - 张量的shape为
。数据类型为float。原子 和 之间的距离 。mask (Tensor) - 张量的shape为
。数据类型为bool。距离 的掩码。默认值:None
。数据类型为bool。距离coulomb_const (float) - 库仑常量
。默认值:1
- 返回:
Tensor。E_coulomb。张量的shape为
。数据类型为float。- 支持平台:
Ascend
GPU