apply

open override fun apply(tf: Ops, yPred: Operand<Float>, yTrue: Operand<Float>, numberOfLosses: Operand<Float>?): Operand<Float>
Content copied to clipboard

Applies LossFunction to the yPred labels predicted by the model and known yTrue hidden during training.

Parameters

yPred

The predicted values. shape = [batch_size, d0, .. dN].

yTrue

Ground truth values. Shape = [batch_size, d0, .. dN], except sparse loss functions such as sparse categorical crossentropy where shape = [batch_size, d0, .. dN-1].

tf

TensorFlow graph API for building operations.