mindspore.mint.dot
- mindspore.mint.dot(input, other)[源代码]
计算两个1DTensor的点积。
警告
这是一个实验性API,后续可能修改或删除。
- 参数:
input (Tensor) - 点积的第一个输入, 须为1D。
other (Tensor) - 点积的第二个输入,须为1D。
- 返回:
Tensor,shape是[], 类型与input一致。
- 异常:
TypeError - input 和 other 的数据类型不是tensor。
TypeError - input 或 other 的数据类型不是float16, float32, 或bfloat16。
RuntimeError - input 和 other 的数据类型不一致。
RuntimeError - input 和 other 的shape不一致。
RuntimeError - input 或 other 不是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