Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

tfds.features.TensorInfo

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
)

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.

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'

__eq__

View source

__eq__(
    other
)

Equality.