用pytorch 实现张量相加
时间: 2023-07-17 22:57:10 浏览: 102
pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
使用PyTorch实现张量相加也非常简单。您可以使用`torch.add()`函数或直接使用`+`操作符进行张量相加。以下是一个示例:
```python
import torch
tensor1 = torch.tensor([[1, 2, 3], [4, 5, 6]])
tensor2 = torch.tensor([[7, 8, 9], [10, 11, 12]])
# 使用torch.add()函数
result1 = torch.add(tensor1, tensor2)
print(result1)
# 使用+操作符
result2 = tensor1 + tensor2
print(result2)
```
输出:
```
tensor([[ 8, 10, 12],
[14, 16, 18]])
tensor([[ 8, 10, 12],
[14, 16, 18]])
```
希望这可以帮助您使用PyTorch实现张量相加。如果您有任何其他问题,请随时问我。
阅读全文