sponge.function.calc_angle_by_vectors

查看源文件
sponge.function.calc_angle_by_vectors(vector1: Tensor, vector2: Tensor, keepdims: bool = False)[源代码]

计算两个向量之间的夹角。对于向量 V1=(x1,x2,x3,...,xn) 和向量 V2=(y1,y2,y3,...,yn) ,两向量间夹角计算公式为:

θ=arccos|x1y1+x2y2++xnyn|x12+x22++xn2y12+y22++yn2
参数:
  • vector1 (Tensor) - 向量1,shape为 (...,D) ,数据类型为float。其中 D 为模拟系统的维度,一般为3。

  • vector2 (Tensor) - 向量2,shape为 (...,D) ,数据类型为float。

  • keepdims (bool) - 如果设置为True,则在结果中,最后一个轴将保留为大小为1的维度。默认值: False

返回:

Tensor。计算所得角。shape为 (...,1) ,数据类型为float。

支持平台:

Ascend GPU