torch.dot

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

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

与Numpy的点不同，torch.dot只支持计算两个元素数量相同的1维张量的点积。

input（张量）–点积中的第一个张量，必须是1维。

other（张量）-点积中的第二个张量，必须是1维。

out（张量，可选）–输出张量。

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