Сложение тензоров в PyTorch
Два тензора одной формы можно сложить
оператором +: в каждой ячейке
результата будет сумма чисел из той
же позиции обоих операндов.
Создадим два ряда и сложим их:
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
c = a + b
print(c) # выведет tensor([5, 7, 9])
Таблицы складывают так же, если число рядов и столбцов совпадает:
import torch
x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[10, 20], [30, 40]])
z = x + y
print(z) # выведет tensor([[11, 22], [33, 44]])
Соберите два ряда [2, 4, 6]
и [1, 1, 1], получите их
сумму по ячейкам и выведите её.
Соберите таблицы [[0, 1], [2, 3]]
и [[5, 5], [5, 5]], получите
сумму по ячейкам и выведите её.
Соберите ряд [10, 20, 30]
и ряд [3, 7, 11], получите
сумму по ячейкам и выведите её.