tf.RaggedTensorSpec

TensorFlow 1 version View source on GitHub

Type specification for a tf.RaggedTensor.

tf.RaggedTensorSpec(
    shape=None, dtype=tf.dtypes.float32, ragged_rank=None,
    row_splits_dtype=tf.dtypes.int64
)

shape The shape of the RaggedTensor, or None to allow any shape. If a shape is specified, then all ragged dimensions must have size None.
dtype tf.DType of values in the RaggedTensor.
ragged_rank Python integer, the ragged rank of the RaggedTensor to be described. Defaults to shape.ndims - 1.
row_splits_dtype dtype for the RaggedTensor's row_splits tensor. One of tf.int32 or tf.int64.

value_type The Python type for values that are compatible with this TypeSpec.

Methods

from_value

View source

@classmethod
from_value(
    value
)

is_compatible_with

View source

is_compatible_with(
    spec_or_value
)

Returns true if spec_or_value is compatible with this TypeSpec.

most_specific_compatible_type

View source

most_specific_compatible_type(
    other
)

Returns the most specific TypeSpec compatible with self and other.

Args
other A TypeSpec.

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

__eq__

View source

__eq__(
    other
)

Return self==value.

__ne__

View source

__ne__(
    other
)

Return self!=value.