Missed TensorFlow World? Check out the recap. Learn more

tf.estimator.export.ExportOutput

TensorFlow 2.0 version View source on GitHub

Class ExportOutput

Represents an output of a model that can be served.

Aliases:

  • Class tf.compat.v1.estimator.export.ExportOutput
  • Class tf.compat.v2.estimator.export.ExportOutput

These typically correspond to model heads.

Methods

as_signature_def

View source

as_signature_def(receiver_tensors)

Generate a SignatureDef proto for inclusion in a MetaGraphDef.

The SignatureDef will specify outputs as described in this ExportOutput, and will use the provided receiver_tensors as inputs.

Args:

  • receiver_tensors: a Tensor, or a dict of string to Tensor, specifying input nodes that will be fed.