sponge.function

接口名	概述	支持平台
sponge.function.calc_angle_by_vectors	计算两个向量之间的夹角。	Ascend GPU
sponge.function.calc_angle_nopbc	计算非周期性边界条件下三个空间位点A，B，C所形成的角度 $\mathrm{\angle }ABC$。	Ascend GPU
sponge.function.calc_angle_pbc	计算周期性边界条件下三个空间位点A，B，C所形成的角度 $\mathrm{\angle }ABC$。	Ascend GPU
sponge.function.calc_angle	计算三个空间位点A，B，C所形成的角度 $\mathrm{\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
sponge.function.all_none	如果 iterable 中的 ALL 值 x 为 None，返回True。	Ascend GPU CPU
sponge.function.all_not_none	如果 iterable 中的所有值 x 不为 None，返回True。	Ascend GPU CPU
sponge.function.any_none	如果可迭代对象中的ANY值 x 为 None，则返回 True。	Ascend GPU CPU
sponge.function.any_not_none	如果 iterable 中的任一值 x 不为 None，返回True。	Ascend GPU CPU
sponge.function.bonds_in	如果 bonds 中存在 batch_bond 则返回。	Ascend GPU CPU
sponge.function.check_broadcast	检查两个shapes是否符合播出规则。	Ascend GPU CPU
sponge.function.concat_first_dim	在第一根轴上连接张量(axis=0)。	Ascend GPU CPU
sponge.function.concat_last_dim	在最后一根轴上连接张量(axis=-1)。	Ascend GPU CPU
sponge.function.concat_penulti	在倒数第二根轴上连接张量(axis=-2)。	Ascend GPU CPU
sponge.function.coordinate_in_pbc	在主PBC box中获取坐标。	Ascend GPU
sponge.function.coulomb_interaction	计算库仑相互作用。	Ascend GPU
sponge.function.get_arguments	获取类的参数。	Ascend GPU CPU
sponge.function.get_energy	获取特定单位的能量张量。	Ascend GPU CPU
sponge.function.get_initializer	按名称获取初始值设定项。	Ascend GPU CPU
sponge.function.get_length	获取特定单位的长度张量。	Ascend GPU CPU
sponge.function.get_ms_array	获取输入值的mindspore.Tensor类型。	Ascend GPU CPU
sponge.function.get_tensor	获取输入值的mindspore.Tensor类型。	Ascend GPU CPU
sponge.function.hamiltonian_product	获取给定四元数和张量的哈密顿乘积。	Ascend GPU CPU
sponge.function.identity	返回与输入具有相同shape和内容的张量。	Ascend GPU CPU
sponge.function.keepdims_mean	对轴中从维度为 x 减小到维度为 1 的元素求平均值，并且输出和输入的维度相同。	Ascend GPU CPU
sponge.function.keepdims_prod	对轴中从维度为 x 减小到维度为 1 的元素相乘，并且输出和输入的维度相同。	Ascend GPU CPU
sponge.function.keepdims_sum	对轴中从维度为 x 减小到维度为 1 的元素求和，并且输出和输入的维度相同。	Ascend GPU CPU
sponge.function.lennard_jones_potential	使用 $ϵ$ 和 $\sigma$ 计算Lennard-Jones (LJ) 势。	Ascend GPU
sponge.function.lennard_jones_potential2	使用 $ϵ$ 和 $r_0$ 计算Lennard-Jones (LJ) 势。	Ascend GPU
sponge.function.periodic_difference	获取周期变量之间的差值。	Ascend GPU
sponge.function.periodic_variable	获取周期范围内的值。	Ascend GPU
sponge.function.quaternion_inverse	获取给定张量的四元数共轭。	Ascend GPU CPU
sponge.function.quaternion_multiply	对给定张量进行四元数乘法操作。	Ascend GPU CPU
sponge.function.reduce_all	默认情况下，通过维度中所有元素的 "逻辑与" 来减少张量的维度。	Ascend GPU CPU
sponge.function.reduce_any	默认情况下，通过维度中所有元素的 "逻辑或" 来减少张量的维度。	Ascend GPU CPU
sponge.function.reduce_prod	默认情况下，通过乘以维度中所有元素来减少张量的维度。	Ascend GPU CPU
sponge.function.squeeze_first_dim	返回从第一个轴(axis=0)删除大小为1的维度后的张量。	Ascend GPU CPU
sponge.function.squeeze_last_dim	返回从最后一个轴(axis=-1)删除大小为1的维度后的张量。	Ascend GPU CPU
sponge.function.squeeze_penulti	返回从倒数第二个轴(axis=-2)删除大小为1的维度后的张量。	Ascend GPU CPU
sponge.function.stack_first_dim	堆叠第一根轴上的张量列表(axis=0)。	Ascend GPU CPU
sponge.function.stack_last_dim	堆叠最后一根轴上的张量列表(axis=-1)。	Ascend GPU CPU
sponge.function.stack_penulti	堆叠倒数第二根轴上的张量列表(axis=-2)。	Ascend GPU CPU