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 )
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 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 . |
Attributes | |
---|---|
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. |
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.3/api_docs/python/tf/RaggedTensorSpec