Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tff.learning.BatchOutput

View source

Class BatchOutput

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

NOTE: All fields are optional (may be None).

  • 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).

Properties

loss

predictions