pytorch 向量乘法
时间: 2023-10-18 13:25:45 浏览: 123
基于pytorch 预训练的词向量用法详解
4星 · 用户满意度95%
在 PyTorch 中,可以使用 `torch.mul()` 函数或者 `*` 运算符来进行向量乘法。
使用 `torch.mul()` 函数的示例如下:
```python
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = torch.mul(a, b)
print(result)
```
输出结果为:
```
tensor([ 4, 10, 18])
```
使用 `*` 运算符的示例如下:
```python
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
result = a * b
print(result)
```
输出结果同样为:
```
tensor([ 4, 10, 18])
```
这两种方法都可以实现向量的逐元素乘法。
阅读全文