тензорный поток:: опс:: Собирать
#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) |
Публичные атрибуты | |
---|---|
operation | |
output |
Общественные функции | |
---|---|
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 )