tff.learning.BatchOutput

Class BatchOutput

Defined in learning/model.py.

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.
  • predictions: Tensor of predictions on the examples. The first dimension must be the same size (the size of the batch).

__new__

__new__(
    _cls,
    loss,
    predictions
)

Create new instance of BatchOutput(loss, predictions)

Properties

loss

predictions