Google I/O returns May 18-20! Reserve space and build your schedule Register now

tff.learning.BatchOutput

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

tff.learning.BatchOutput(
    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.

loss Instance of member_descriptor
num_examples Instance of member_descriptor
predictions Instance of member_descriptor