PyTorch torch.frexp 関数


Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

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)

出力結果は次のとおりです:

尾数: tensor([0.5000, 0.5625, 0.5000])
指数: tensor([4, 3, 2])
验证: tensor([8., 4.5, 2.])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張