публичный финальный класс SplitV
Разбивает тензор на тензоры `num_split` по одному измерению.
Публичные методы
static <T, U расширяет число> SplitV <T> | create (область области действия , значение операнда <T>, операнд <U> sizeSplits, ось операнда <Integer>, Long numSplit) Фабричный метод для создания класса, обертывающего новую операцию SplitV. |
Итератор< Операнд <T>> | итератор () |
Список< Выход <T>> | выход () Тензоры, форма которых совпадает с формой значения, за исключением вдоль оси, где их размеры равны size_splits[i]`. |
Унаследованные методы
Публичные методы
public static SplitV <T> create (область действия , значение операнда <T>, операнд <U> sizeSplits, ось операнда <Integer>, Long numSplit)
Фабричный метод для создания класса, обертывающего новую операцию SplitV.
Параметры
объем | текущий объем |
---|---|
ценить | Тензор для разделения. |
размерРазделение | список, содержащий размеры каждого выходного тензора по разделенному измерению. Сумма должна соответствовать размеру значения вдоль Split_dim. Может содержать один -1, указывающий, что размер должен быть выведен. |
ось | 0-Д. Измерение, по которому выполняется разделение. Должно быть в диапазоне `[-rank(значение), Rank(значение))`. |
Возврат
- новый экземпляр SplitV
общедоступный список < Вывод <T>> вывод ()
Тензоры, форма которых совпадает с формой значения, за исключением вдоль оси, где их размеры равны size_splits[i]`.