Возведение в степень в PyTorch

Чтобы возвести элементы в степень, используют оператор **. Если справа тоже тензор той же формы, степень берётся по ячейкам; одно число справа применяют ко всем элементам слева.

Создадим два ряда и возведём первый в степени второго:

import torch

base = torch.tensor([2, 3, 4])
exp = torch.tensor([2, 2, 2])
out = base ** exp
print(out) # выведет tensor([ 4,  9, 16])

Одну и ту же степень можно задать числом для каждого элемента ряда:

import torch

t = torch.tensor([1, 2, 3, 4])
u = t ** 2
print(u) # выведет tensor([ 1,  4,  9, 16])

Соберите ряд [2, 3, 5] и ряд [3, 2, 1], возведите первый в степени второго и выведите результат.

Соберите ряд [1, 2, 3, 4], поднимите каждый его элемент в квадрат и выведите получившийся ряд.

Соберите таблицу [[2, 3], [4, 5]] и таблицу [[1, 2], [2, 1]], получите степень по ячейкам и выведите её.