Функция torch.pow в PyTorch


Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.powЭто функция в PyTorch, используемая для вычисления поэлементных операций возведения в степень.

Определение функции

torch.pow(input, exponent, out)

Пример использования

Пример

import torch

# Вычислить операцию возведения в степень
x = torch.tensor([2, 3, 4])
result = torch.pow(x, 2)
print(result)

# Вы также можете использовать показатель степени в качестве первого аргумента
result2 = torch.pow(2, x)
print(result2)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения