공개 최종 클래스 XlaConcatND
모든 차원에 걸쳐 입력 텐서를 연결합니다.
병합하는 작업은 주어진 num_splits 속성을 기반으로 입력 텐서를 분할하고, 선택적으로 패딩을 제거하고, 패딩 없이 병합된 텐서를 반환합니다.
이 작업은 TPU 브리지를 통해 생성될 수 있습니다.
예를 들어 `입력` 텐서를 사용하면 다음과 같습니다.
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
[2, 2]
[1, 1]
[[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
중첩 클래스
수업 | XlaConcatND.옵션 | Xla 의 선택적 속성 |
공개 방법
출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
정적 <T> XlaConcatND <T> | 생성 ( Scope 범위, Iterable< Operand <T>> 입력, List<Long> numConcats, Options... 옵션) 새로운 XlaConcatND 작업을 래핑하는 클래스를 생성하는 팩터리 메서드입니다. |
출력 <T> | 출력 () |
정적 XlaConcatND.옵션 | 패딩 (List<Long> 패딩) |
상속된 메서드
부울 | 같음 (개체 arg0) |
마지막 수업<?> | getClass () |
정수 | 해시코드 () |
최종 무효 | 알림 () |
최종 무효 | 통지모두 () |
끈 | toString () |
최종 무효 | 대기 (long arg0, int arg1) |
최종 무효 | 기다리세요 (긴 arg0) |
최종 무효 | 기다리다 () |
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
public static XlaConcatND <T> create ( Scope 범위, Iterable< Operand <T>> 입력, List<Long> numConcats, Options... 옵션)
새로운 XlaConcatND 작업을 래핑하는 클래스를 생성하는 팩터리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
입력 | 모든 차원에 걸쳐 병합하려면 행 우선 순서로 텐서 슬라이스를 입력하세요. 모든 입력은 동일한 형태를 가져야 합니다. } out_arg { 이름: "출력" 설명: < |
numConcats | 차원당 병합하는 방법의 수입니다. |
옵션 | 선택적 속성 값을 전달합니다. |
보고
- XlaConcatND의 새 인스턴스
공개 정적 XlaConcatND.Options 패딩 (List<Long> 패딩)
매개변수
패딩 | 최종 병합된 텐서에서 제거할 차원당 오른쪽 패딩의 선택적 목록입니다. 이러한 패딩은 패딩을 제거하기 전에 병합된 결과의 차원 크기를 초과해서는 안 됩니다. |
---|