XlaSplitND

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

Разбивает входной тензор по всем измерениям.

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

Эта операция может быть сгенерирована через мост TPU.

Например, для тензора `input`:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 

`num_splits`:

[2, 2]
 

и `paddings`:

[1, 1]
 

ожидаемые `выходные данные`:

[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

Вложенные классы

сорт

XlaSplitND.Параметры

Дополнительные атрибуты для XlaSplitND

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

статический <T> XlaSplitND <T>	create ( Область действия , ввод операнда <T>, Long N, List<Long> numSplits, параметры... ) Фабричный метод для создания класса, обертывающего новую операцию XlaSplitND.
Итератор< Операнд <T>>	итератор ()
Список< Выход <T>>	выходы ()
статический XlaSplitND.Options	отступы (отступы List<Long>)

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

Из класса org.tensorflow.op.PrimitiveOp

последнее логическое значение	равно (Объект obj)
окончательный интервал	хэш-код ()
Операция	оп () Возвращает базовую Operation
последняя строка	нанизывать ()

Из класса java.lang.Object

логическое значение	равно (Объект arg0)
последний класс<?>	получитьКласс ()
интервал	хэш-код ()
окончательная пустота	поставить в известность ()
окончательная пустота	уведомитьВсе ()
Нить	нанизывать ()
окончательная пустота	подождать (длинный arg0, int arg1)
окончательная пустота	подождите (длинный arg0)
окончательная пустота	ждать ()

Из интерфейса java.lang.Iterable

пустота	forEach (Consumer<? super T> arg0)
абстрактный Итератор< Операнд <T>>	итератор ()
Разделитель< Операнд <T>>	сплитатор ()

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

public static XlaSplitND <T> create (область действия , ввод операнда <T>, Long N, List<Long> numSplits, параметры... параметры)

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

Параметры

объем	текущий объем
вход	Введите тензор для разделения по всем измерениям. } out_arg { name: "outputs" описание: <
numSplits	Количество способов разделения по измерению. Размеры фигуры должны делиться без остатка.
параметры	содержит значения необязательных атрибутов

Возврат

• новый экземпляр XlaSplitND

общедоступный итератор < Операнд <T>> итератор ()

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

общедоступные статические дополнения XlaSplitND.Options (дополнения List<Long>)

Параметры

прокладки	Необязательный список правых заполнений для каждого измерения входного тензора, которые необходимо применить перед разделением. Это можно использовать для того, чтобы сделать размерность делимой без остатка.