sponge.function

接口名

概述

支持平台

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