221 Star 947 Fork 694

GVPMindSpore/mindscience

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

sponge.function.calc_angle

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

    计算三个空间位点A,B,C所形成的角度 :math:`\angle ABC`。

    如果提供盒子尺寸pbc_box,则按照周期性边界条件下的坐标进行计算。如果盒子尺寸pbc_box为None,则按照非周期性边界条件下的坐标进行计算。

    最后返回 :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。默认值: ``None`` 。
        - **keepdims** (bool) - 如果设置为True,则在结果中,最后一个轴将保留为大小为1的维度。默认值:``False``。

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

搜索帮助