XlaSpmdShardToFullShape

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

Операция, используемая средством разделения XLA SPMD для переключения с ручного разделения на

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

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

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

Выход <Т>
какВывод ()
Возвращает символический дескриптор тензора.
static <T расширяет TType > XlaSpmdShardToFullShape <T>
create (область действия , ввод операнда <T>, строка manualSharding, форма fullShape)
Фабричный метод для создания класса, обертывающего новую операцию XlaSpmdShardToFullShape.
Выход <Т>

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

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «XlaSpmdShardToFullShape»

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

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

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

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

public static XlaSpmdShardToFullShape <T> create (область области действия , ввод операнда <T>, строка manualSharding, форма fullShape)

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

Параметры
объем текущий объем
Возврат
  • новый экземпляр XlaSpmdShardToFullShape

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