mindspore.Tensor.dot

查看源文件
mindspore.Tensor.dot(other)

计算两个1DTensor的点积。

参数:
  • other (Tensor) - 点积的输入,须为1D。

返回:

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

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

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

  • RuntimeError - selfother 的shape不一致。

  • RuntimeError - selfother 不是1D。

支持平台:

Ascend

样例:

>>> import mindspore
>>> from mindspore import Tensor
>>> input = Tensor([2.0, 3.0], mindspore.float32)
>>> other = Tensor([2.0, 1.0], mindspore.float32)
>>> output = Tensor.dot(input, other)  # input.dot(other)
>>> print(output)
[7.        ]
>>> print(output.dtype)
Float32