torch中张量的乘法
时间: 2023-09-22 11:09:35 浏览: 100
pytorch-extension:一个使用CuPy的PyTorch的CUDA扩展示例,该示例计算两个张量的Hadamard乘积
在PyTorch中,可以使用`torch.mul()`函数或者直接使用`*`运算符来执行张量的乘法操作。这两种方法都可以用于两个张量的逐元素乘法,即对应位置上的元素相乘。
下面是使用这两种方法执行张量乘法的示例代码:
```python
import torch
# 创建两个张量
tensor1 = torch.tensor([1, 2, 3])
tensor2 = torch.tensor([4, 5, 6])
# 方法一:使用torch.mul()函数
result1 = torch.mul(tensor1, tensor2)
print(result1) # 输出: tensor([ 4, 10, 18])
# 方法二:使用*运算符
result2 = tensor1 * tensor2
print(result2) # 输出: tensor([ 4, 10, 18])
```
这两种方法执行的操作效果是相同的,都是对应位置上的元素进行相乘。
阅读全文