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.3/api_docs/python/tf/raw_ops/RaggedGather