Skip to content

.forward()

Compute cross-entropy loss.

Parameters:

  • z (Tensor) –

    Logits tensor.

  • y (Tensor) –

    Target probability distribution, same shape as z.