Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tff.learning.BatchOutput

View source on GitHub

A structure that holds the output of a tff.learning.Model.

@staticmethod
tff.learning.BatchOutput(
    _cls, loss, predictions, num_examples
)

Used in the notebooks

Used in the tutorials
  • loss: The scalar mean loss on the examples in the batch. If the model has multiple losses, it is the sum of all the individual losses.
  • predictions: Tensor of predictions on the examples. The first dimension must be the same size (the size of the batch).
  • num_examples: Number of examples seen in the batch.

Attributes:

  • loss
  • predictions
  • num_examples