torch.linalg.vecdot¶
- torch.linalg.vecdot(x, y, *, dim=-1, out=None) Tensor ¶
沿一个维度计算两个向量批次的点积。
用符号表示,此函数计算
在维度
dim
上,其中 表示复数向量的共轭,对于实数向量则为恒等式。支持 half、bfloat16、float、double、cfloat、cdouble 和整数数据类型作为输入。它也支持广播。
- 参数
- 关键字参数
示例
>>> v1 = torch.randn(3, 2) >>> v2 = torch.randn(3, 2) >>> linalg.vecdot(v1, v2) tensor([ 0.3223, 0.2815, -0.1944]) >>> torch.vdot(v1[0], v2[0]) tensor(0.3223)