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型インスタンス