mindspore.numpy.inner

mindspore.numpy.inner(a, b)[源代码]

返回两个Tensor的内积。

对于1-D的Tensor，这是向量的普通内积(不包含复共轭)。对于更高维的Tensor，这是在最后一个轴上的求和积。

说明

不支持NumPy参数 out。在GPU上，支持的数据类型有np.float16和np.float32。在CPU上，支持的数据类型有np.float16，np.float32和np.float64。

参数：

a (Tensor) - 输入Tensor。如果 a 和 b 都不是标量，它们的最后一个维度必须匹配。
b (Tensor) - 输入Tensor。如果 a 和 b 都不是标量，它们的最后一个维度必须匹配。

返回：

Tensor或标量。

异常：

ValueError - 如果 x1.shape[-1] != x2.shape[-1] 。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> a = np.ones((5, 3))
>>> b = np.ones((2, 7, 3))
>>> output = np.inner(a, b)
>>> print(output)
[[[3. 3. 3. 3. 3. 3. 3.]
[3. 3. 3. 3. 3. 3. 3.]]
[[3. 3. 3. 3. 3. 3. 3.]
[3. 3. 3. 3. 3. 3. 3.]]
[[3. 3. 3. 3. 3. 3. 3.]
[3. 3. 3. 3. 3. 3. 3.]]
[[3. 3. 3. 3. 3. 3. 3.]
[3. 3. 3. 3. 3. 3. 3.]]
[[3. 3. 3. 3. 3. 3. 3.]
[3. 3. 3. 3. 3. 3. 3.]]]