tf.data.DatasetSpec

Type specification for tf.data.Dataset.

Inherits From: TypeSpec, TraceType

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([]))

element_specThe inner element spec.
value_typeThe 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

experimental_as_proto

View source

Returns a proto representation of the TypeSpec instance.

Do NOT override for custom non-TF types.

experimental_from_proto

View source

Returns a TypeSpec instance based on the serialized proto.

Do NOT override for custom non-TF types.

Args
protoProto generated using 'experimental_as_proto'.

experimental_type_proto

View source

Returns the type of proto associated with TypeSpec serialization.

Do NOT override for custom non-TF types.

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.

Prefer using "is_subtype_of" and "most_specific_common_supertype" wherever possible.

Args
spec_or_valueA TypeSpec or TypeSpec associated value to compare against.

is_subtype_of

View source

See base class.

most_specific_common_supertype

View source

See base class.

most_specific_compatible_type

View source

Returns the most specific TypeSpec compatible with self and other. (deprecated)

Deprecated. Please use most_specific_common_supertype instead. Do not override this function.

Args
otherA TypeSpec.

Raises
ValueErrorIf there is no TypeSpec that is compatible with both self and other.

__eq__

View source

Return self==value.

__ne__

View source

Return self!=value.