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`.

© 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/raw_ops/RaggedGather