将两个张量相加
时间: 2023-09-21 20:08:02 浏览: 140
第3章_深度学习_
可以使用PyTorch的`add()`函数将两个张量相加,代码如下:
```python
import torch
# 创建两个5x3的随机初始化张量
x = torch.randn(5, 3)
y = torch.randn(5, 3)
print("张量x:")
print(x)
print("张量y:")
print(y)
# 将x与y相加
z = torch.add(x, y)
print("张量z:")
print(z)
```
输出结果如下:
```
张量x:
tensor([[-0.0966, 0.6181, -0.5511],
[-0.4646, 0.6139, 0.0910],
[-0.1404, -0.9431, -0.4493],
[ 0.6127, 0.0474, -1.1644],
[-0.4741, -0.0123, 1.0516]])
张量y:
tensor([[ 0.5517, -0.8669, 0.6939],
[-1.2213, -0.3117, -0.8766],
[-1.4354, -1.0666, -0.0788],
[-0.0150, 0.3649, -0.1745],
[ 1.3071, 0.4449, -0.7775]])
张量z:
tensor([[ 0.4551, -0.2488, 0.1428],
[-1.6860, 0.3022, -0.7856],
[-1.5758, -2.0098, -0.5281],
[ 0.5977, 0.4123, -1.3389],
[ 0.8329, 0.4326, 0.2741]])
```
在这个例子中,我们首先创建了两个5x3的随机初始化张量`x`和`y`,然后使用`add()`函数将它们相加生成了一个新的张量`z`。注意,`add()`函数的第一个参数表示要相加的第一个张量,第二个参数表示要相加的第二个张量。
阅读全文