Reihenweises Kreuz Entropie Verlust in 3D Tensor

Question

Ich habe ein rekurrentes neuronales Netzwerk, das eine Ausgabe von Form [sequence_length, batch_size, num_classes] ausgibt und ich möchte softmax_cross_entropy_with_logits auf jeden Zeitstempel der Sequenz anwenden, oder auf jede Zeile, wenn Sie (korrekte Etiketten sind die gleiche Form). Wie ist das möglich, ohne über die Sequenzlänge zu iterieren, die wirklich langsam sein kann?

Und wie geht es mit Genauigkeit, wenn wir dabei sind?

Answer 1

Offenbar ist Broadcast in der Lage, damit gut umzugehen. Ich hatte Probleme wegen falsch formatierter Logits und Labels.