用Pytorch实现dotproduct
时间: 2023-08-15 12:39:06 浏览: 64
py-distributed-dot-product:基于MPI的点产品关注度分布式实现的PyTorch实现
可以使用`torch.matmul`函数来实现两个张量的点积(dot product),如下所示:
```python
import torch
# 创建两个张量
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
# 计算点积
dot_product = torch.matmul(a, b)
print(dot_product)
```
输出结果为:
```
tensor(32)
```
其中,`torch.matmul`函数会根据张量的维度自动选择是执行矩阵乘法还是元素乘积并相加。在本例中,由于输入的都是1维张量,因此执行的是元素乘积并相加,得到了点积的结果。
阅读全文