공개 최종 수업 SplitV
텐서를 하나의 차원을 따라 `num_split` 텐서로 분할합니다.
공개 방법
static <T, U 확장 번호> SplitV <T> | |
반복자< 피연산자 <T>> | 반복자 () |
목록< 출력 <T>> | 출력 () 크기가 `size_splits[i]`인 `축`을 제외하고 모양이 `값`과 일치하는 텐서입니다. |
상속된 메서드
공개 방법
공개 정적 SplitV <T> 생성 ( 범위 범위, 피연산자 <T> 값, 피연산자 <U> sizeSplits, 피연산자 <Integer> 축, Long numSplit)
새로운 SplitV 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
값 | 분할할 텐서입니다. |
크기분할 | 분할 차원에 따른 각 출력 텐서의 크기를 포함하는 목록입니다. 분할_dim을 따라 값의 차원을 합산해야 합니다. 차원이 추론됨을 나타내는 하나의 -1을 포함할 수 있습니다. |
중심선 | 0-D. 분할할 차원입니다. '[-순위(값), 순위(값))' 범위에 있어야 합니다. |
보고
- SplitV의 새 인스턴스