Q&A
PyTorchで手書き数字（０〜９）の判別のためのCNNモデルを構築しています。
下記のコードで出力結果が異なる理由を教えて頂けますでしょうか？
また、最初のコードで2つ目のコードと同じ出力にするにはどのように変更すれば良いのでしょうか？
python
1#y -> ０〜９各数字に対するモデルの出力値 2#t -> 各画像の正解ラベル 3cross_ent1 = -(torch.log_softmax(y[torch.arange(batch_size),t],dim=-1)).sum().mean()
python
1y_soft = torch.nn.functional.softmax(y, dim=-1) 2cross_ent2 = -(torch.log(y_soft[np.arange(batch_size),t])).sum().mean()
分かる方いらっしゃいましたらよろしくお願い致します。
あなたの回答
tips
プレビュー
