カード編集

問題

答え・判断基準 クラス方向なので axis=1。shape維持が必要なら keepdims=True。

種類 即答 選択肢消去 数式 PyTorch対応 用語

章

論点

解説 (N,C)のNはバッチサイズ(サンプル数)、Cはクラス数を表す。Softmaxは「各サンプルの中で」クラス方向に確率を正規化する必要があるため、axis=1(列方向、Cの軸)を指定する。axis=0を指定するとバッチ内の異なるサンプル同士を混ぜて集計してしまい誤り。ブロードキャストで割り算する際に形状を保つためkeepdims=Trueも忘れずに。

Is active