import torch.nn as nn


def cross_entropy_loss():
    """Create Cross Entropy Loss.
    The loss automatically applies the softmax activation
    function on the prediction input.

    Returns:
        Cross entroy loss function
    """
    return nn.CrossEntropyLoss()

c26e8a40ebfb3a17acd3ffd9f093ebef7cbfe94f
<mask token>

<mask token>


def cross_entropy_loss():
    """Create Cross Entropy Loss.
    The loss automatically applies the softmax activation
    function on the prediction input.

    Returns:
        Cross entroy loss function
    """
    return nn.CrossEntropyLoss()

import torch.nn as nn


def cross_entropy_loss():
    """Create Cross Entropy Loss.
    The loss automatically applies the softmax activation
    function on the prediction input.

    Returns:
        Cross entroy loss function
    """
    return nn.CrossEntropyLoss()
