sponge.function

接口名

概述

支持平台

sponge.function.calc_angle_by_vectors

计算两个向量之间的夹角。

Ascend GPU

sponge.function.calc_angle_nopbc

计算非周期性边界条件下三个空间位点A,B,C所形成的角度 ABC

Ascend GPU

sponge.function.calc_angle_pbc

计算周期性边界条件下三个空间位点A,B,C所形成的角度 ABC

Ascend GPU

sponge.function.calc_angle

计算三个空间位点A,B,C所形成的角度 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×box0.5×box 的范围内。

Ascend GPU

sponge.function.VelocityGenerator

根据温度产生系统中原子的速度。

Ascend GPU CPU