tf.data.DatasetSpec

Type specification for tf.data.Dataset.

See tf.TypeSpec for more information about TensorFlow type specifications.

dataset = tf.data.Dataset.range(3)
tf.data.DatasetSpec.from_value(dataset)
DatasetSpec(TensorSpec(shape=(), dtype=tf.int64, name=None), TensorShape([]))

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.

Methods

from_value

View source

Creates a DatasetSpec for the given tf.data.Dataset value.

is_compatible_with

View source

Returns true if spec_or_value is compatible with this TypeSpec.

most_specific_compatible_type

View source

Returns the most specific TypeSpec compatible with self and other.

Args
other A TypeSpec.

Raises