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)

这将输出点乘的结果，即 `32`。