espnet2.speechlm.net_utils.ce_loss

source

espnet2.speechlm.net_utils.ce_loss(logits: Tensor, target: Tensor, lengths: Tensor, prefix_len: Tensor | None = None, first_layer_weight: int = 1.0) → Tuple[Tensor, Tensor]