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
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	使用 \(\epsilon\) 和 \(\sigma\) 计算Lennard-Jones (LJ) 势。	Ascend GPU
sponge.function.lennard_jones_potential2	使用 \(\epsilon\) 和 \(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