pytorch 使用点积

在PyTorch中，可以使用torch.dot()函数来计算两个张量的点积。点积也称为内积或标量积，是两个向量相乘后再求和的结果。

以下是使用点积的示例代码：

import torch

# 创建两个张量
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])

# 计算点积
dot_product = torch.dot(a, b)

print(dot_product)

输出结果为：

tensor(32)

上述代码中，我们首先导入了torch模块，然后创建了两个张量a和b。接下来，使用torch.dot()函数计算了a和b的点积，并将结果存储在dot_product变量中。最后，我们打印了dot_product的值。