Обертывает оператор XLA Gather, описанный по адресу
https://www.tensorflow.org/xla/operation_semantics#gather
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
static <T расширяет TType , U расширяет TNumber > Gather <T> | |
Выход <Т> | выход () |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static Gather <T> create (область области действия , операнд <T> операнд, операнд <U> startIndices, операнд <U> слайсы, строковые измерения, логические индексыAreSorted)
Фабричный метод для создания класса, обертывающего новую операцию Gather.
Параметры
объем | текущий объем |
---|---|
операнд | Массив, из которого мы собираем данные. |
startIndices | Массив, содержащий начальные индексы собранных нами срезов. |
размеры срезов | срез_размеры[i] — это границы среза в измерении i. |
размерностьЧисла | Сериализованный прототип xla::GatherDimensionNumbers. |
индексыAreSorted | Логическое значение, указывающее, отсортированы ли индексы. |
Возврат
- новый экземпляр Gather