tensorflow::ops::GatherV2

#include <array_ops.h>

Gather срезы из params оси axis в соответствии с indices.

Краткое описание

indices должна быть целочисленным тензором любой размерности (обычно 0-й или 1-й). Создаёт выходной тензор с формой 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.

Аргументы:

• scope: Объект Scope
• params: Тензор, из которого извлекаются значения. Должен иметь размерность не менее axis + 1.
• indices: Тензор индексов. Должен находиться в диапазоне [0, params.shape[axis]).
• axis: Ось в params для извлечения indices по заданным индексам. По умолчанию — первая ось. Поддерживаются отрицательные индексы.

Возвращает:

• Output: Значения из params извлечённые по индексам, указанным в indices, с формой params.shape[:axis] + indices.shape + params.shape[axis + 1:].

Общедоступные атрибуты

Operation operation

::tensorflow::Output output

Общедоступные функции

 GatherV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  ::tensorflow::Input axis
)

 GatherV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  ::tensorflow::Input axis,
  const GatherV2::Attrs & attrs
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const 

Общедоступные статические функции

Attrs BatchDims(
  int64 x
)