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