python - テンソルフローの sparse_softmax_cross_entropy_with_logits 関数の元のコーディングはどこですか

Question

テンソルフロー関数sparse_softmax_cross_entropy_with_logitsが数学的に正確に何をしているのか知りたいです。しかし、コーディングの起源を見つけることができません。手伝って頂けますか？

Answer 1

実装の最も重要な部分は、ここの132 行目からです。

このファンクターは、カーネルの実装によって呼び出されます。

ジェネレーターと呼ばれる Eigen のあまり文書化されていない機能を使用して、かなり柔軟なコードを記述し、CPU 用と GPU 用の nvcc の両方でコンパイルできるようにします。

Answer 2

ヘッド バージョン (現在) では、 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/nn_ops.py行 #424で関数を見つけることができます。

コメントには次のように書かれています。

クラスが相互に排他的である (各エントリが厳密に 1 つのクラスにある) 離散分類タスクの確率誤差を測定します。たとえば、各 CIFAR-10 画像は 1 つのラベルだけでラベル付けされます。画像は犬またはトラックのいずれかである場合がありますが、両方である場合はありません。