PyTorch torch.pow 関数
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)
# べき乗演算を計算する
x = torch.tensor([2, 3, 4])
result = torch.pow(x, 2)
print(result)
# 指数を最初の引数として使用することもできます
result2 = torch.pow(2, x)
print(result2)
その他の拡張