Class BatchOutput
A structure that holds the output of a
tff.learning.Model.
Defined in
python/learning/model.py.
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).