torch.dot¶
- torch.dot(input, tensor, *, out=None) Tensor ¶
计算两个 1D 张量的点积。
注意
与 NumPy 的 dot 不同,torch.dot 仅支持计算具有相同元素数量的两个 1D 张量的点积。
- 参数
- 关键字参数
out (Tensor, 可选) – 输出张量。
示例
>>> torch.dot(torch.tensor([2, 3]), torch.tensor([2, 1])) tensor(7) >>> t1, t2 = torch.tensor([0, 1]), torch.tensor([2, 3]) >>> torch.dot(t1, t2) tensor(3)