tfds.features.TensorInfo

View source on GitHub

Structure containing info on the tf.Tensor shape/dtype.

tfds.features.TensorInfo(
    shape: tfds.typing.Shape,
    dtype: tf.dtypes.DType,
    default_value=None,
    sequence_rank: tfds.typing.Dim = None,
    dataset_lvl: int = 0
)

Args
shape	tuple[int], shape of the tensor
dtype	Tensor dtype
default_value	Used for retrocompatibility with previous files if a new field is added to provide a default value when reading the file.
sequence_rank	int, Number of tfds.features.Sequence dimension.
dataset_lvl	int, if >0, nesting level of a tfds.features.Dataset.

Attributes
dataset_lvl
default_value
dtype
numpy_dtype
sequence_rank
shape

Methods

copy_from

View source

@classmethod
copy_from(
    tensor_info: 'TensorInfo'
) -> 'TensorInfo'

Copy constructor.

from_tensor_spec

View source

@classmethod
from_tensor_spec(
    tensor_spec: tf.TensorSpec
) -> 'TensorInfo'

to_tensor_spec

View source

to_tensor_spec() -> tf.TensorSpec

Converts this TensorInfo instance to a tf.TensorSpec.

Note that there is a bug (b/227584124) around RaggedTensorSpec, so the output for sequences of sequences may not be correct.

Returns
The tf.TensorSpec corresponding to this instance.

__eq__

View source

__eq__(
    other
)

Equality.