210 Star 890 Fork 659

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
sponge.function.calc_angle_pbc.rst 1.36 KB
一键复制 编辑 原始数据 按行查看 历史
Li Qingguo 提交于 2023-10-12 20:14 . fix docs

sponge.function.calc_angle_pbc

.. py:function:: sponge.function.calc_angle_pbc(position_a: Tensor, position_b: Tensor, position_c: Tensor, pbc_box: Tensor, keepdims: bool = False)

    计算周期性边界条件下三个空间位点A,B,C所形成的角度 :math:`\angle ABC`。输入A,B,C三点坐标与pbc_box,返回夹角 :math:`\angle ABC` 大小。

    根据周期性边界条件下A,B,C三点坐标计算向量 :math:`\vec{BA}` 和 :math:`\vec{BC}` 的坐标,再计算两向量间夹角。

    最后返回 :math:`\vec{BA}` 向量与 :math:`\vec{BC}` 向量间夹角。

    参数:
        - **position_a** (Tensor) - 位置a,shape为 :math:`(B, ..., D)` ,数据类型为float。其中B为Batch size,D为模拟系统的维度, 一般为3。
        - **position_b** (Tensor) - 位置b,shape为 :math:`(B, ..., D)` ,数据类型为float。
        - **position_c** (Tensor) - 位置c,shape为 :math:`(B, ..., D)` ,数据类型为float。
        - **pbc_box** (Tensor) - PBC box,shape为 :math:`(B, D)` ,数据类型为float。
        - **keepdims** (bool) - 如果设置为True,则在结果中,最后一个轴将保留为大小为1的维度。默认值: ``False``。

    返回:
        Tensor。计算所得角。shape为 :math:`(B, ..., 1)` ,数据类型为float。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
master

搜索帮助