dot

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

计算向量的内积

注解

支持 1 维和 2 维 Tensor。如果是 2 维 Tensor,矩阵的第一个维度是 batch_size,将会在多个样本上进行点积计算。

参数

  • x (Tensor)- 1 维或 2 维 Tensor。数据类型为 float32float64int32int64

  • y (Tensor)- 1 维或 2 维 Tensor。数据类型为 float32float64int32int64

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

返回

Tensor,数据类型与 x 相同。

代码示例

import paddle

# 1-D Tensor * 1-D Tensor
x = paddle.to_tensor([1, 2, 3])
y = paddle.to_tensor([4, 5, 6])
z = paddle.dot(x, y)
print(z)  # [32]

# 2-D Tensor * 2-D Tensor
x = paddle.to_tensor([[1, 2, 3], [2, 4, 6]])
y = paddle.to_tensor([[4, 5, 6], [4, 5, 6]])
z = paddle.dot(x, y)
print(z)  # [[32], [64]]