Softmax関数【torch】
説明:
指定されたベクトルが有する値を、範囲が
$ 0~1
の値に変換する関数である。値の総和は1であるので、確率値として解釈することができる。
$ y_i(\mathbf{x}) = \frac{\exp(x_i)}{\sum_{j=1}^{N}\exp(x_j)}
書式:
torch.nn.Softmax( dim )
引数
dim:
ソフトマックス処理を適用する
配列の軸
の向きを指定する名前付き引数。
sumメソッド【numpy】
における
axis
に相当する。
戻り値
Softmax型インスタンス