接口名 |
概述 |
支持平台 |
sponge.function.calc_angle_by_vectors
|
计算两个向量之间的夹角。 |
Ascend GPU
|
sponge.function.calc_angle_nopbc
|
计算非周期性边界条件下三个空间位点A,B,C所形成的角度 \(\angle ABC\)。 |
Ascend GPU
|
sponge.function.calc_angle_pbc
|
计算周期性边界条件下三个空间位点A,B,C所形成的角度 \(\angle ABC\)。 |
Ascend GPU
|
sponge.function.calc_angle
|
计算三个空间位点A,B,C所形成的角度 \(\angle ABC\)。 |
Ascend GPU
|
sponge.function.calc_distance_nopbc
|
在没有周期性边界条件的情况下计算位置A和B之间的距离,用绝对坐标计算。 |
Ascend GPU
|
sponge.function.calc_distance_pbc
|
在有周期性边界条件的情况下计算位置A和B之间的距离,需要转化为同一个 pbc_box 内坐标计算 A 和B 的距离。 |
Ascend GPU
|
sponge.function.calc_distance
|
计算位置A和B之间的距离。 |
Ascend GPU
|
sponge.function.calc_torsion_by_vectors
|
计算由三个向量形成的扭转角。 |
Ascend GPU
|
sponge.function.calc_torsion_nopbc
|
在没有周期性边界条件的情况下计算由四个位置A,B,C,D形成的扭转角。 |
Ascend GPU
|
sponge.function.calc_torsion_pbc
|
在有周期性边界条件的情况下计算由四个位置A,B,C,D形成的扭转角。 |
Ascend GPU
|
sponge.function.calc_torsion
|
计算由四个位置A,B,C,D形成的扭转角。 |
Ascend GPU
|
sponge.function.calc_vector_nopbc
|
在没有周期性边界条件的情况下,计算从起点到终点的向量。 |
Ascend GPU
|
sponge.function.calc_vector_pbc
|
在有周期性边界条件的情况下,计算从起点到终点的向量。 |
Ascend GPU
|
sponge.function.calc_vector
|
计算从起点到终点的向量。 |
Ascend GPU
|
sponge.function.energy_convert
|
根据不同的单位换算能量。 |
Ascend GPU CPU
|
sponge.function.Energy
|
能量。 |
Ascend GPU CPU
|
sponge.function.gather_value
|
根据指标从张量的最后一根轴收集值。 |
Ascend GPU
|
sponge.function.gather_vector
|
根据指标从张量的倒数第二轴收集向量。 |
Ascend GPU
|
sponge.function.get_energy_ref
|
获得参考能量。 |
Ascend GPU CPU
|
sponge.function.get_energy_unit_name
|
获得能量单位的名称。 |
Ascend GPU CPU
|
sponge.function.get_energy_unit
|
获得能量单位。 |
Ascend GPU CPU
|
sponge.function.get_integer
|
获取输入的整数类型。 |
Ascend GPU CPU
|
sponge.function.get_length_ref
|
获得参考长度。 |
Ascend GPU CPU
|
sponge.function.get_length_unit_name
|
获得长度单位的名称。 |
Ascend GPU CPU
|
sponge.function.get_length_unit
|
获得长度单位。 |
Ascend GPU CPU
|
sponge.function.get_ndarray
|
获取输入的ndarray类型。 |
Ascend GPU CPU
|
sponge.function.GetDistance
|
获取有或者没有PBC box的距离。 |
Ascend GPU CPU
|
sponge.function.GetDistanceShift
|
计算维数为C的B矩阵。 |
Ascend GPU CPU
|
sponge.function.GetShiftGrad
|
计算维度为 \((K, N, D)\) 的B矩阵的微分,N是原子的邻居原子数,D是模拟系统的维度。 |
Ascend GPU CPU
|
sponge.function.GetVector
|
获取有或者没有PBC box的向量。 |
Ascend GPU CPU
|
sponge.function.length_convert
|
根据不同的单位换算长度。 |
Ascend GPU CPU
|
sponge.function.Length
|
获取长度。 |
Ascend GPU CPU
|
sponge.function.pbc_box_reshape
|
把PBC box改变shape,使它的维度与ndim相同。 |
Ascend GPU
|
sponge.function.pbc_image
|
计算PBC box的周期图。 |
Ascend GPU
|
sponge.function.set_global_length_unit
|
给全局单位设置长度单位。 |
Ascend GPU CPU
|
sponge.function.set_global_energy_unit
|
给全局单位设置能量单位。 |
Ascend GPU CPU
|
sponge.function.set_global_units
|
给全局单位设置单位。 |
Ascend GPU CPU
|
sponge.function.Units
|
记录和转换长度和能量单位。 |
Ascend GPU CPU
|
sponge.function.vector_in_pbc
|
在周期性边界条件下,使向量在 \(-0.5 \times box\) 到 \(0.5 \times box\) 的范围内。 |
Ascend GPU
|
sponge.function.VelocityGenerator
|
根据温度产生系统中原子的速度。 |
Ascend GPU CPU
|