PyTorch torch.rsqrt 関数
Pytorch torch リファレンスマニュアルtorch.rsqrtPyTorch において、要素ごとの平方根の逆数を計算するための関数です。
関数定義
torch.rsqrt(input, out=None)
使用例
実例
import torch
# 平方根の逆数を計算
x = torch.tensor([1.0, 4.0, 9.0])
result = torch.rsqrt(x)
print(result)
# 平方根の逆数を計算
x = torch.tensor([1.0, 4.0, 9.0])
result = torch.rsqrt(x)
print(result)
その他の拡張