tf.raw_ops.GatherV2

Собирает срезы из params оси axis согласно indices.

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.GatherV2

tf.raw_ops.GatherV2(
    params, indices, axis, batch_dims=0, name=None
)

indices должен быть целочисленным тензором любой размерности (обычно 0-D или 1-D). Создает выходной тензор с формой params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:], где:

# Scalar indices (output is rank(params) - 1).
output[a_0, ..., a_n, b_0, ..., b_n] =
  params[a_0, ..., a_n, indices, b_0, ..., b_n]

# Vector indices (output is rank(params)).
output[a_0, ..., a_n, i, b_0, ..., b_n] =
  params[a_0, ..., a_n, indices[i], b_0, ..., b_n]

# Higher rank indices (output is rank(params) + rank(indices) - 1).
output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
  params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]

Обратите внимание: на CPU при обнаружении индекса, выходящего за границы, возвращается ошибка. На GPU при обнаружении индекса, выходящего за границы, в соответствующее значение выходного тензора записывается 0.

См. также tf.batch_gather и tf.gather_nd.

Аргументы
`params`	Тензор `Tensor`. Тензор, из которого извлекаются значения. Его ранг должен быть не меньше `axis + 1`.
`indices`	Тензор `Tensor`. Должен иметь один из следующих типов: `int16`, `int32`, `int64`. Тензор индексов. Значения должны находиться в диапазоне `[0, params.shape[axis])`.
`axis`	Тензор `Tensor`. Должен иметь один из следующих типов: `int32`, `int64`. Ось в `params`, из которой извлекаются `indices`. По умолчанию используется первое измерение. Поддерживаются отрицательные индексы.
`batch_dims`	Необязательный `int`. По умолчанию — `0`.
`name`	Имя операции (необязательно).

Возвращаемое значение
Тензор `Tensor`. Имеет тот же тип, что и `params`.

© 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/GatherV2