import torch


def get_accuracy_with_logits(y_true: torch.Tensor, y_pred: torch.Tensor) -> float:
    """
    Calculate accuracy with logits
    :param y_true: torch.Tensor of ints containing true labels. shape = (N,)
    :param y_pred: torch.Tensor of predicted logits. shape = (N, N_classes)
    :return acc: float which equals to accuracy score
    """

    return (y_true == y_pred.argmax(dim=1)).float().mean().item()