inner

paddle. inner ( x, y, name=None ) [源代码]

计算两个 Tensor 的内积。

对于 1 维 Tensor 计算普通内积,对于大于 1 维的 Tensor 计算最后一个维度的乘积和,此时两个输入 Tensor 最后一个维度长度需要相等。

参数

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

代码示例:

>>> import paddle
>>> x = paddle.arange(1, 7).reshape((2, 3)).astype('float32')
>>> y = paddle.arange(1, 10).reshape((3, 3)).astype('float32')
>>> out = paddle.inner(x, y)
>>> print(out)
Tensor(shape=[2, 3], dtype=float32, place=Place(cpu), stop_gradient=True,
[[14. , 32. , 50. ],
 [32. , 77. , 122.]])