SoftMax

バージョン名: SoftMax-8

カテゴリー: 活性化関数

簡単な説明: リファレンス

詳細な説明: リファレンス

属性

• axis

  • 説明: axis は、SoftMax が計算される axis を表します。負の値は、次元を後ろから数えることを意味します。axis に等しい 1 がデフォルト値です。

  • 値の範囲: [-rank, rank - 1]

  • タイプ: int

  • デフォルト値: 1

  • 必須: いいえ

数学定式化

\[y_{c} = \frac{e^{Z_{c}}}{\sum_{d=1}^{C}e^{Z_{d}}}\]

ここで \(C\) は axis 次元に沿ったテンソルのサイズです。

入力:

• 1: axis 属性と互換性がある次元数の入力テンソル。必須。

出力:

• 1: 入力テンソルと同じ形状およびタイプの結果として得られるテンソル。

例

<layer ... type="SoftMax"...> 
    <data axis="1" /> 
        <input>
             ...       </Input> 
        <output>
             ...        </output> 
</layer>