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

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


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

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

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

torch.float_power(input, exponent, *, out=None)

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

Пример

import torch

# Пример операции возведения в степень с плавающей запятой
x = torch.tensor([2.0, 3.0, 4.0])
y = torch.tensor([2.0, 2.0, 2.0])
result = torch.float_power(x, y)
print(result)  # tensor([4., 9., 16.])

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

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