mindspore.ops.tensor_dot

mindspore.ops.tensor_dot(x1, x2, axes)

沿指定轴计算两个张量的点乘。

参数：

• x1 (Tensor) - 输入Tensor。

• x2 (Tensor) - 输入Tensor。

• axes (Union[int, tuple(int), tuple(tuple(int)), list(list(int))]) - 要求和的维度数。如果是整数 k ，则对 x1 的后 k 个轴和 x2 的前 k 个轴求和。如果提供的是列表或元组，则axes[0] 指定 x1 的轴，axes[1] 指定 x2 的轴。

返回：

Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> from mindspore import Tensor, ops
>>> import mindspore
>>> import numpy as np
>>> input_x1 = Tensor(np.ones(shape=[1, 2, 3]), mindspore.float32)
>>> input_x2 = Tensor(np.ones(shape=[3, 1, 2]), mindspore.float32)
>>> output = ops.tensor_dot(input_x1, input_x2, ((0,1),(1,2)))
>>> print(output)
[[2. 2. 2]
 [2. 2. 2]
 [2. 2. 2]]