Gather

публичный финальный класс Gather

Соберите срезы из оси «params» в соответствии с «индексами».

`индексы` должны быть целочисленными тензорами любой размерности (обычно 0-D или 1-D). Создает выходной тензор формы `params.shape[:axis] + index.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]

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

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

Вложенные классы

сорт Собрать.Параметры Дополнительные атрибуты для Gather

Публичные методы

Выход <Т>
какВывод ()
Возвращает символический дескриптор тензора.
статический Gather.Options
батчдимс (длинный батчдимс)
static <T, U расширяет число, V расширяет число> Собрать <T>
create ( Область действия , параметры операнда <T>, индексы операнда <U>, ось операнда <V>, параметры ... )
Фабричный метод для создания класса, обертывающего новую операцию Gather.
Выход <Т>
выход ()
Значения из `params`, собранные из индексов, заданных `indices`, с формой `params.shape[:axis] + index.shape + params.shape[axis + 1:]`.

Унаследованные методы

Публичные методы

публичный вывод <T> asOutput ()

Возвращает символический дескриптор тензора.

Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.

общедоступный статический Gather.Options пакетный Dims (длинный пакетный Dims)

public static Gather <T> create (область действия , параметры операнда <T>, индексы операнда <U>, ось операнда <V>, параметры... параметры)

Фабричный метод для создания класса, обертывающего новую операцию Gather.

Параметры
объем текущий объем
параметры Тензор, из которого собираются значения. Должен быть как минимум рангом `axis + 1`.
индексы Индексный тензор. Должно быть в диапазоне `[0, params.shape[axis])`.
ось Ось в params, из которой собираются индексы. По умолчанию используется первое измерение. Поддерживает отрицательные индексы.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр Gather

публичный вывод <T> вывод ()

Значения из `params`, собранные из индексов, заданных `indices`, с формой `params.shape[:axis] + index.shape + params.shape[axis + 1:]`.