mindspore.ops.vecdot

mindspore.ops.vecdot(x, y, *, axis=- 1)

按指定轴，计算两批向量的点积。

支持广播。

计算公式如下， 如果 x 是复数向量，$\overline{x_i}$ 表示向量中元素的共轭；如果 x 是实数向量，$\overline{x_i}$ 表示向量中元素本身。

$$\sum _{i=1}^n\overline{x_i}{y}_i$$

警告

这是一个实验性API，后续可能修改或删除。

参数：

• x (Tensor) - 第一批tensor。

• y (Tensor) - 第二批tensor。

关键字参数：

• axis (int) - 指定计算轴。默认 -1 。

返回：

Tensor

说明

当前在GPU上不支持复数。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([[1, 3], [5, 7], [9, 8]])
>>> y = mindspore.tensor([[4, 5], [6, 7], [3, 2]])
>>> mindspore.ops.vecdot(x, y)
Tensor(shape=[3], dtype=Int64, value= [19, 79, 43])
>>> mindspore.ops.vecdot(x, y, axis=0)
Tensor(shape=[2], dtype=Int64, value= [61, 80])