tf.RaggedTensorSpec
| Просмотреть исходный код на GitHub |
Спецификация типа для tf.RaggedTensor.
Наследуется от: TypeSpec
tf.RaggedTensorSpec(
shape=None, dtype=tf.dtypes.float32, ragged_rank=None,
row_splits_dtype=tf.dtypes.int64, flat_values_spec=None
)
| Аргументы | |
|---|---|
shape | Форма RaggedTensor, или None для поддержки любой формы. Если форма указана, все разрозненные измерения должны иметь размер None. |
dtype | tf.DType значений в RaggedTensor. |
ragged_rank | Целое число Python, количество раз, когда плоские значения RaggedTensor разбиваются. По умолчанию shape.ndims - 1. |
row_splits_dtype | dtype для тензора row_splits RaggedTensor. Один из tf.int32 или tf.int64. |
flat_values_spec | TypeSpec для плоского значения RaggedTensor. Он должен быть предоставлен, когда flat_values является CompositeTensor, а не Tensor. Если и dtype и flat_values_spec предоставлены, dtype должно совпадать с flat_values_spec.dtype. (экспериментально) |
| Атрибуты | |
|---|---|
dtype | tf.dtypes.DType, заданный этим типом для RaggedTensor. rt = tf.ragged.constant([["a"], ["b", "c"]], dtype=tf.string) tf.type_spec_from_value(rt).dtype tf.string |
flat_values_spec | Форма плоских значений RaggedTensor. |
ragged_rank | Количество раз, когда плоские значения RaggedTensor разбиваются. По умолчанию values = tf.ragged.constant([[1, 2, 3], [4], [5, 6], [7, 8, 9, 10]]) tf.type_spec_from_value(values).ragged_rank 1 rt1 = tf.RaggedTensor.from_uniform_row_length(values, 2) tf.type_spec_from_value(rt1).ragged_rank 2 |
row_splits_dtype | tf.dtypes.DType тензора row_splits RaggedTensor. rt = tf.ragged.constant([[1, 2, 3], [4]], row_splits_dtype=tf.int64) tf.type_spec_from_value(rt).row_splits_dtype tf.int64 |
shape | Статически известная форма RaggedTensor. rt = tf.ragged.constant([[0], [1, 2]]) tf.type_spec_from_value(rt).shape TensorShape([2, None]) rt = tf.ragged.constant([[[0, 1]], [[1, 2], [3, 4]]], ragged_rank=1) tf.type_spec_from_value(rt).shape TensorShape([2, None, 2]) |
value_type | Тип Python для значений, совместимых с этим TypeSpec. В частности, все значения, совместимые с этим TypeSpec, должны быть экземпляром этого типа. |
Методы
from_value
@classmethod
from_value(
value
)
is_compatible_with
is_compatible_with(
spec_or_value
)
Возвращает true, если spec_or_value совместим с этим TypeSpec.
most_specific_compatible_type
most_specific_compatible_type(
other
)
Возвращает наиболее специфичный TypeSpec, совместимый с self и other.
| Аргументы | |
|---|---|
other | TypeSpec. |
| Исключения | |
|---|---|
ValueError | Если нет TypeSpec, который совместим с self и other. |
__eq__
__eq__(
other
)
Возвращает self==value.
__ne__
__ne__(
other
)
Возвращает 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