三维张量中的逐行交叉熵损失

Question

我有一个递归神经网络，它输出形状[sequence_length, batch_size, num_classes]的输出，我想将softmax_cross_entropy_with_logits应用于序列的每个时间戳，或者如果你愿意，则应用于每一行（正确的标签是相同的形状）。如果没有迭代可能非常慢的序列长度，这怎么可能呢？

我们在这里时如何做到准确？

Answer 1

显然，广播能够很好地处理这个问题。我因为错误格式化的logits和标签而遇到了问题。