Функция torch.exp2 в PyTorch
Справочное руководство Pytorch torch
torch.exp2Это функция в PyTorch, используемая для поэлементного вычисления степени 2.
Определение функции
torch.exp2(input, *, out=None)
Пример использования
Пример
import torch
# Пример возведения в степень 2
x = torch.tensor([0, 1, 2, 3, 4], dtype=torch.float32)
result = torch.exp2(x)
print(result) # tensor([1., 2., 4., 8., 16.])
# Пример возведения в степень 2
x = torch.tensor([0, 1, 2, 3, 4], dtype=torch.float32)
result = torch.exp2(x)
print(result) # tensor([1., 2., 4., 8., 16.])
Другие расширения