tf.raw_ops.RaggedGather

Извлечение разрозненных срезов из params по оси 0 согласно indices.

Просмотр псевдонимов

Псевдонимы для миграции

Для получения дополнительной информации см. Руководство по миграции.

tf.compat.v1.raw_ops.RaggedGather

tf.raw_ops.RaggedGather(
    params_nested_splits,
    params_dense_values,
    indices,
    OUTPUT_RAGGED_RANK,
    name=None
)

Выводит RaggedTensor результат, составленный из output_dense_values и output_nested_splits, такой что:

output.shape = indices.shape + params.shape[1:]
output.ragged_rank = indices.shape.ndims + params.ragged_rank
output[i...j, d0...dn] = params[indices[i...j], d0...dn]

где

params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits) предоставляет значения, которые должны быть извлечены.
indices — это плотный тензор с типом int32 или int64, указывающий, какие значения должны быть извлечены.
output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits) — это выходной тензор.

(Примечание: этот C++ оператор используется для реализации оператора Python более высокого уровня tf.ragged.gather , который также поддерживает разрозненные индексы.)

Аргументы
`params_nested_splits`	Список из как минимум 1 `Tensor` объектов одного типа в: `int32`, `int64`. Тензоры `nested_row_splits`, определяющие разделение строк для входного `params` RaggedTensor.
`params_dense_values`	`Tensor`. `flat_values` для `params` RaggedTensor. В Python произошла смена терминологии с dense_values на flat_values, поэтому dense_values — устаревшее название.
`indices`	`Tensor`. Должен быть одним из следующих типов: `int32`, `int64`. Индексы во внешнем измерении `params` значений, которые должны быть извлечены.
`OUTPUT_RAGGED_RANK`	`int` , являющийся `>= 0`. Разрозненность ранга выходного RaggedTensor. `output_nested_splits` будет содержать это количество `row_splits` тензоров. Это значение должно быть равно `indices.shape.ndims + params.ragged_rank - 1`.
`name`	Название операции (необязательно).

Возвращаемые значения
Кортеж из `Tensor` объектов (output_nested_splits, output_dense_values).
`output_nested_splits`	Список `OUTPUT_RAGGED_RANK` `Tensor` объектов того же типа, что и `params_nested_splits`.
`output_dense_values`	`Tensor`. Имеет тот же тип, что и `params_dense_values`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/raw_ops/RaggedGather