On this page
tf.RaggedTensorSpec
Type specification for a tf.RaggedTensor.
Inherits From: TypeSpec
tf.RaggedTensorSpec(
shape=None, dtype=tf.dtypes.float32, ragged_rank=None,
row_splits_dtype=tf.dtypes.int64, flat_values_spec=None
)
| Args | |
|---|---|
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 number of times the RaggedTensor's flat_values is partitioned. Defaults to shape.ndims - 1. |
row_splits_dtype |
dtype for the RaggedTensor's row_splits tensor. One of tf.int32 or tf.int64. |
flat_values_spec |
TypeSpec for flat_value of the RaggedTensor. It shall be provided when the flat_values is a CompositeTensor rather then Tensor. If both dtype and flat_values_spec and are provided, dtype must be the same as flat_values_spec.dtype. (experimental) |
| Attributes | |
|---|---|
dtype |
The tf.dtypes.DType specified by this type for the RaggedTensor. |
flat_values_spec |
The TypeSpec of the flat_values of RaggedTensor. |
ragged_rank |
The number of times the RaggedTensor's flat_values is partitioned. Defaults to |
row_splits_dtype |
The tf.dtypes.DType of the RaggedTensor's row_splits. |
shape |
The statically known shape of the RaggedTensor. |
value_type |
The 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
from_value
@classmethod
from_value(
value
)
is_compatible_with
is_compatible_with(
spec_or_value
)
Returns true if spec_or_value is compatible with this TypeSpec.
most_specific_compatible_type
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__
__eq__(
other
)
Return self==value.
__ne__
__ne__(
other
)
Return self!=value.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/RaggedTensorSpec