import torch


def softmax_output_to_label(y: torch.Tensor) -> torch.Tensor:
    """
    Args:
        y: 3D tensor (batch_size, output_size, seq_len)
           output_size = n_classes

    Returns:
        y: 2D tensor (batch_size, seq_len)
    """
    return y.argmax(dim=1)