Split

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

Разбивает тензор на тензоры `num_split` по одному измерению.

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

статический <T> Разделить <T>
create (область действия , ось операнда <Integer>, значение операнда <T>, Long numSplit)
Фабричный метод для создания класса, обертывающего новую операцию разделения.
Итератор< Операнд <T>>
Список< Выход <T>>
выход ()
Это тензоры одинаковой формы, чья форма совпадает с формой `value`, за исключением вдоль `axis`, где их размеры равны `values.shape[split_dim]/num_split`.

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

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

public static Split <T> create (область действия , ось операнда <Integer>, значение операнда <T>, Long numSplit)

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

Параметры
объем текущий объем
ось 0-Д. Измерение, по которому выполняется разделение. Должно быть в диапазоне `[-rank(значение), Rank(значение))`.
ценить Тензор для разделения.
numSplit Количество способов разделения. Необходимо равномерно разделить `value.shape[split_dim]`.
Возврат
  • новый экземпляр Сплита

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

общедоступный список < Вывод <T>> вывод ()

Это тензоры одинаковой формы, чья форма совпадает с формой `value`, за исключением вдоль `axis`, где их размеры равны `values.shape[split_dim]/num_split`.