Операция разбивает кортеж входных данных дедупликации XLA на целочисленные тензоры и тензоры с плавающей запятой.
Данные дедупликации представляют собой кортеж XLA, который состоит из целых чисел и значений с плавающей запятой. Эта операция состоит в том, чтобы разделить эти значения на две группы для двух типов и построить каждую группу как один тензор для возврата.
Вложенные классы
сорт | SplitDedupData.Параметры | Дополнительные атрибуты для SplitDedupData |
Публичные методы
статический SplitDedupData.Options | конфигурация (строковая конфигурация) |
static <T расширяет число, U расширяет число> SplitDedupData <T, U> | create ( Область области , ввод операнда <?>, класс <T> целочисленный тип, класс <U> floatType, строка tupleMask, параметры... параметры) Фабричный метод для создания класса, обертывающего новую операцию SplitDedupData. |
Выход <U> | ФлотТензор () Одномерный тензор с плавающей запятой включает элементы с плавающей запятой кортежа данных дедупликации. |
Выход <Т> | целоеТензор () Одномерный целочисленный тензор включает в себя целочисленные элементы кортежа данных дедупликации. |
Унаследованные методы
Публичные методы
public static SplitDedupData <T, U> create (область области действия , ввод операнда <?>, Class<T> IntegerType, Class<U> floatType, String tupleMask, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию SplitDedupData.
Параметры
объем | текущий объем |
---|---|
вход | Кортеж XLA, включающий целочисленные элементы и элементы с плавающей точкой в качестве кортежа данных дедупликации. |
целоетипе | тип целое_тензор. Разрешенные типы: int32, int64, uint32, uint64. |
тип плавающего типа | Тип float_tensor. Разрешенные типы: half, bfloat16, float. |
кортежМаска | Сериализованная строка TensorProto маски выходного кортежа. Эта маска представляет собой двумерный тензор, в котором первый столбец представляет собой тип элемента кортеж, а второй столбец — диапазон этого типа. Например, выходной кортеж (1, 2, 0,1, 3), его маска — [[0, 2], [1, 1], [0, 1]]. Мы ожидаем только два типа элементов: целое число (0) и число с плавающей запятой (1). |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр SplitDedupData
публичный вывод <U> floatTensor ()
Одномерный тензор с плавающей запятой включает элементы с плавающей запятой кортежа данных дедупликации.
публичный вывод <T> целоеTensor ()
Одномерный целочисленный тензор включает в себя целочисленные элементы кортежа данных дедупликации.