dot
计算向量的内积
备注
支持 1 维和 2 维 Tensor。如果是 2 维 Tensor,矩阵的第一个维度是 batch_size,将会在多个样本上进行点积计算。
备注
别名支持: 参数名 input
可替代 x
, tensor
可替代 y
。
参数
x (Tensor) - 1 维或 2 维
Tensor
。数据类型为float32
、float64
、int32
、int64
、complex64
或complex128
。 别名:input
y (Tensor) - 1 维或 2 维
Tensor
。数据类型为float32
、float64
、int32
、int64
、complex64
或complex128
。 别名:tensor
name (str,可选) - 具体用法请参见 api_guide_Name,一般无需设置,默认值为 None。
关键字参数
out (Tensor,可选) - 指定输出结果的 Tensor,默认值为 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)
Tensor(shape=[], dtype=int64, place=Place(cpu), stop_gradient=True,
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)
Tensor(shape=[2], dtype=int64, place=Place(cpu), stop_gradient=True,
[32, 64])