публичный финальный класс Сплит
Разбивает тензор на тензоры `num_split` по одному измерению.
Публичные методы
статический <T> Разделить <T> | |
Итератор< Операнд <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>> вывод ()
Это тензоры одинаковой формы, чья форма соответствует форме `value`, за исключением вдоль `axis`, где их размеры равны `values.shape[split_dim]/num_split`.