PyTorch torch.i0 関数
torch.i0PyTorch の torch.i0 は、位数0の第一種修正ベッセル関数(Modified Bessel Function of the First Kind)を計算する関数です。
関数定義
torch.i0(input)
使用例
例
import torch
# 第一種修正ベッセル関数を計算する
x = torch.tensor([0.0, 1.0, 2.0, 3.0])
y = torch.i0(x)
print("I0(x):", y)
# 第一種修正ベッセル関数を計算する
x = torch.tensor([0.0, 1.0, 2.0, 3.0])
y = torch.i0(x)
print("I0(x):", y)
出力結果は次のとおりです:
I0(x): tensor([1.0000, 1.2661, 2.2796, 4.8808])
その他の拡張
Pytorch torch リファレンスマニュアル