mindspore.mint.dot

查看源文件
mindspore.mint.dot(input, other)[源代码]

计算两个1DTensor的点积。

警告

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

参数:
  • input (Tensor) - 点积的第一个输入, 须为1D。

  • other (Tensor) - 点积的第二个输入,须为1D。

返回:

Tensor,shape是[], 类型与input一致。

异常:
  • TypeError - inputother 的数据类型不是tensor。

  • TypeError - inputother 的数据类型不是float16, float32, 或bfloat16。

  • RuntimeError - inputother 的数据类型不一致。

  • RuntimeError - inputother 的shape不一致。

  • RuntimeError - inputother 不是1D。

支持平台:

Ascend

样例:

>>> import mindspore
>>> from mindspore import Tensor, mint
>>> x = Tensor([2.0, 3.0], mindspore.float32)
>>> y = Tensor([2.0, 1.0], mindspore.float32)
>>> dot = mint.dot()
>>> output = dot(x, y)
>>> print(output)
[7.        ]
>>> print(output.dtype)
Float32