tff.learning.reconstruction.BatchOutput

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

predictions A tf.Tensor of predictions on the examples.
labels A tf.Tensor of labels for the examples.
num_examples A tf.int32 scalar number of examples seen in the batch.