Число и тензор в PyTorch
Одно число Python можно прибавить к тензору или умножить на него: операция выполняется с каждым элементом, форма тензора не меняется.
Создадим ряд и прибавим к нему число:
import torch
vec = torch.tensor([1, 2, 3])
plus = vec + 10
print(plus) # выведет tensor([11, 12, 13])
Умножение на число действует так же для каждой ячейки:
import torch
vec = torch.tensor([1, 2, 3])
times = vec * 2
print(times) # выведет tensor([2, 4, 6])
Таблица принимает то же правило:
import torch
mat = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
mat2 = mat + 1.0
print(mat2) # выведет tensor([[2., 3.], [4., 5.]])
Соберите ряд [5, 6, 7],
увеличьте каждый его элемент
на 100 и выведите результат.
Соберите ряд [4, 5, 6],
удвойте каждый его элемент
и выведите получившийся ряд.
Соберите таблицу [[0.0, 1.0], [2.0, 3.0]],
увеличьте каждую ячейку на 0.5
и выведите таблицу.