PyTorch torch.float_power 関数
Pytorch torch リファレンスマニュアルtorch.float_powerPyTorch において、要素ごとに浮動小数点のべき乗演算を行うための関数です。
関数定義
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.])
# 浮動小数点べき乗演算の例
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.])
その他の拡張