Соберите срезы из оси «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> | |
Выход <Т> | выход () Значения из `params`, собранные из индексов, заданных `indices`, с формой `params.shape[:axis] + index.shape + params.shape[axis + 1:]`. |
Унаследованные методы
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
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:]`.