フォワード損失ソフトマックス・クロスエントロピー層

問題ステートメント

以下の項目を指定します。

• サイズ n_k x n の 2 次元 (2D) テンソル X ' として表される p 次元の入力テンソル X。ここで、n = n₁ · ... · n_k-1 · n_k+1 · ... · n_p

• p 次元の正解データテンソル T ∈ R ^{1 x ... x n_k x... x 1}

• サイズ n_k x n の 2D テンソル S ' として表されるソフトマックス補助値のテンソル S ∈ R^{n₁ x ... x n_k x... x n_p}

次のような 1 次元のテンソル Y ∈ R ¹ を計算します。

ライブラリーは、s_ij の値の計算に数値的に安定した式を使用します。