PyTorchでテンソルの各要素の対数を計算する方法：torch.log()の使い方

PyTorchでテンソルの各要素の対数（自然対数）を計算するには、torch.log()メソッドを使用します。このメソッドは、入力テンソルの各要素に対する自然対数の値を持つ新しいテンソルを返します。引数としてテンソルを受け取り、結果としてテンソルを出力します。

手順

• 必要なライブラリをインポートします。以下のすべてのPythonサンプルではtorchライブラリを使用します。事前にインストール済みであることを確認してください。

• テンソルを作成し、内容を表示します。

• torch.log(input)を実行します。引数inputにはテンソルを渡し、その各要素の自然対数値を含む新しいテンソルが返されます。

• 元の入力テンソルの各要素の自然対数値を含むテンソルを表示します。

例1：1次元テンソルの場合

次のPythonプログラムは、PyTorchの1次元テンソルに対して自然対数を計算する方法を示しています。

# 必要なライブラリをインポート
import torch

# テンソルを作成
t = torch.Tensor([2.3,3,2.3,4,3.4])

# 作成したテンソルを表示
print("Original tensor:\n", t)

# テンソルの各要素の対数を計算
log = torch.log(t)

# 計算した対数の値を表示
print("Logarithm of Elements:\n", log)

出力

Original tensor:
    tensor([2.3000, 3.0000, 2.3000, 4.0000, 3.4000])
Logrithm of Elements:
    tensor([0.8329, 1.0986, 0.8329, 1.3863, 1.2238])

例2：2次元テンソルの場合

次のPythonプログラムは、2次元テンソルに対して自然対数を計算する方法を示しています。

# 必要なライブラリをインポート
import torch

# サイズ3x4の乱数テンソルを作成
t = torch.rand(3,4)

# 作成したテンソルを表示
print("Original tensor:\n", t)

# テンソルの各要素の対数を計算
log = torch.log(t)

# 計算した対数の値を表示
print("Logarithm of Elements:\n", log)

出力

Original tensor:
tensor([[0.1245, 0.0448, 0.1176, 0.7607],
        [0.7415, 0.7738, 0.0694, 0.6983],
        [0.8371, 0.6169, 0.3858, 0.8027]])
Logarithm of Elements:
tensor([[-2.0837, -3.1048, -2.1405, -0.2735],
        [-0.2990, -0.2565, -2.6676, -0.3591],
        [-0.1778, -0.4830, -0.9524, -0.2198]])

補足：関連する対数関数

なお、PyTorchには自然対数以外にも以下のような対数計算用の関数が用意されています。

• torch.log10()：常用対数（底10）を計算します。
• torch.log2()：二進対数（底2）を計算します。
• torch.log1p()：(1 + x) の自然対数を高精度に計算します。xが非常に小さい場合に便利です。

用途に応じてこれらの関数を使い分けることで、より柔軟な数値計算が可能になります。