tfds.features.TensorInfo

View source on GitHub

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

tfds.features.TensorInfo(
    shape, dtype, default_value=None, sequence_rank=None, dataset_lvl=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.

Methods

copy_from

View source

@classmethod
copy_from(
    tensor_info
)

Copy constructor.

__eq__

View source

__eq__(
    other
)

Equality.

Class Variables
dataset_lvl	Instance of member_descriptor
default_value	Instance of member_descriptor
dtype	Instance of member_descriptor
sequence_rank	Instance of member_descriptor
shape	Instance of member_descriptor