Softmax кросс-энтропийная потеря для одной партии в python

Question

Я хочу спросить, как реализовать кросс-энтропийную потерю для одной партии в нейронной сети, где уравнение:

это мой код для кросс-энтропии только для одной пример:

def softmax_cross_entropy(y_true, y_pred):
    softmax_cross_entropy_loss_single = - np.sum([y * np.log(x) for x, y in zip(y_pred, y_true)])
    softmax_cross_entropy_grad = y_pred - y_true
    return softmax_cross_entropy_loss, softmax_cross_entropy_grad

как реализовать это для одной партии (используя приведенное выше уравнение)?