PyTorch torch.frexp 関数
torch.frexpPyTorch で浮動小数点数を仮数(mantissa)と指数(exponent)に分解するための関数です。
関数定義
torch.frexp(input)
2つのテンソルを返します:
- mantissa - 仮数部分。範囲は [-0.5, 1)
- exponent - 指数部分。整数です。
使用例
例
import torch
# 浮動小数点テンソルを作成
x = torch.tensor([8.0, 4.5, 2.0])
# 仮数と指数に分解
mantissa, exponent = torch.frexp(x)
print("仮数:", mantissa)
print("指数:", exponent)
# 検証: x = mantissa * 2^exponent
print("検証:", mantissa * 2 ** exponent)
# 浮動小数点テンソルを作成
x = torch.tensor([8.0, 4.5, 2.0])
# 仮数と指数に分解
mantissa, exponent = torch.frexp(x)
print("仮数:", mantissa)
print("指数:", exponent)
# 検証: x = mantissa * 2^exponent
print("検証:", mantissa * 2 ** exponent)
出力結果は次のとおりです:
尾数: tensor([0.5000, 0.5625, 0.5000]) 指数: tensor([4, 3, 2]) 验证: tensor([8., 4.5, 2.])
その他の拡張
Pytorch torch リファレンスマニュアル