тензорный поток:: опс:: Собирать
#include <array_ops.h>
Соберите фрагменты из params
в соответствии с indices
.
Краткое содержание
indices
должны быть целочисленным тензором любой размерности (обычно 0-D или 1-D). Создает выходной тензор с формой indices.shape + params.shape[1:]
где:
# Scalar indices
output[:, ..., :] = params[indices, :, ... :]
# Vector indices
output[i, :, ..., :] = params[indices[i], :, ... :]
# Higher rank indices
output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]
Если indices
являются перестановкой и len(indices) == params.shape[0]
то эта операция соответствующим образом переставит params
.
validate_indices
: УСТАРЕЛО. Если эта операция назначена ЦП, значения в indices
всегда проверяются на предмет соответствия диапазону. Если индексы за пределами границы назначены графическому процессору, они приводят к безопасному, но неопределенному поведению, которое может включать возникновение ошибки.
Аргументы:
- область: объект области.
Возврат:
-
Output
: выходной тензор.
Конструкторы и деструкторы | |
---|---|
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices) | |
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::Attrs & attrs) |
Общественные функции | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Публичные статические функции | |
---|---|
ValidateIndices (bool x) |
Структуры | |
---|---|
tensorflow::ops:: Gather:: Attrs | Дополнительные установщики атрибутов для Gather . |
Публичные атрибуты
операция
Operation operation
выход
::tensorflow::Output output
Общественные функции
Собирать
Gather(
const ::tensorflow::Scope & scope,
::tensorflow::Input params,
::tensorflow::Input indices
)
Собирать
Gather(
const ::tensorflow::Scope & scope,
::tensorflow::Input params,
::tensorflow::Input indices,
const Gather::Attrs & attrs
)
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
Валидироватьиндексы
Attrs ValidateIndices(
bool x
)