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

Type specification for

Inherits From: TypeSpec

For instance, can be used to define a tf.function that takes as an input argument:

  tf.TensorSpec(shape=(), dtype=tf.int32, name=None))])
def square(iterator):
  x = iterator.get_next()
  return x * x
dataset =
iterator = iter(dataset)
tf.Tensor(25, shape=(), dtype=int32)

element_spec A (nested) structure of tf.TypeSpec objects that represents the type specification of the iterator elements.
value_type The Python type for values that are compatible with this TypeSpec.

In particular, all values that are compatible with this TypeSpec must be an instance of this type.



View source


View source

Returns true if spec_or_value is compatible with this TypeSpec.


View source

Returns the most specific TypeSpec compatible with self and other.

other A TypeSpec.

ValueError If there is no TypeSpec that is compatible with both self and other.


View source

Return self==value.