Python PyTorch dot用法及代码示例

本文简要介绍python语言中 torch.dot 的用法。

用法:

torch.dot(input, other, *, out=None) → Tensor

参数：

• input(Tensor) -点积中的第一个张量，必须是一维的。

• other(Tensor) -点积中的第二个张量，必须是一维的。

关键字参数：

out(Tensor,可选的) -输出张量。

计算两个一维张量的点积。

注意

与 NumPy 的点不同，torch.dot 有意仅支持计算具有相同元素数量的两个一维张量的点积。

例子：

>>> torch.dot(torch.tensor([2, 3]), torch.tensor([2, 1]))
tensor(7)