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)
#        ([[14, 32, 50],
#         [32, 77, 122]])