ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


TensorFlow 1 version View source on GitHub

Represents the output of a regression head.

Inherits From: ExportOutput

value a float Tensor giving the predicted values. Required.

ValueError if the value is not a Tensor with dtype tf.float32.




View source

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.

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