Pytorch(二)Tensors的四種加法方式

Pytorch(二)Tensors的四種加法方式

加法

先創建 x和y兩個5行三列的二維張量

import torch

x = torch.linspace(1, 15, 15).view(5, 3)
y = torch.rand(5, 3)
print(x, y, sep='\n')

---

tensor([[ 1.,  2.,  3.],
        [ 4.,  5.,  6.],
        [ 7.,  8.,  9.],
        [10., 11., 12.],
        [13., 14., 15.]])
tensor([[0.0228, 0.6401, 0.9500],
        [0.8969, 0.2857, 0.6268],
        [0.3445, 0.6496, 0.0873],
        [0.0726, 0.7710, 0.1559],
        [0.9020, 0.0102, 0.4305]])

加法形式一

print(x + y)

加法形式二

print(torch.add(x, y))

加法形式三 指定輸出對象

result = torch.empty(5, 3)
torch.add(x, y, out=result)
print(result)

加法形式四 直接改變y inplace

效果及等同於 y = x + y

y.add_(x)
print(y)

---

上面四種方法的加法結果都是一樣的,如下

tensor([[ 1.0228,  2.6401,  3.9500],
        [ 4.8969,  5.2857,  6.6268],
        [ 7.3445,  8.6496,  9.0873],
        [10.0726, 11.7710, 12.1559],
        [13.9020, 14.0102, 15.4305]])