XlaConcatND

공개 최종 클래스 XlaConcatND

모든 차원에 걸쳐 입력 텐서를 연결합니다.

병합하는 작업은 주어진 num_splits 속성을 기반으로 입력 텐서를 분할하고, 선택적으로 패딩을 제거하고, 패딩 없이 병합된 텐서를 반환합니다.

이 작업은 TPU 브리지를 통해 생성될 수 있습니다.

예를 들어 `입력` 텐서: 

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
`num_splits`: 
[2, 2]
및 `paddings`: 
[1, 1]
의 경우 예상되는 `출력`은 
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
입니다.

중첩 클래스

수업 XlaConcatND.옵션 XlaConcatND 의 선택적 속성

공개 방법

출력 <T>
출력 ()
텐서의 기호 핸들을 반환합니다.
정적 <T> XlaConcatND <T>
생성 ( Scope 범위, Iterable< Operand <T>> 입력, List<Long> numConcats, Options... 옵션)
새로운 XlaConcatND 작업을 래핑하는 클래스를 생성하는 팩터리 메서드입니다.
출력 <T>
출력 ()
정적 XlaConcatND.옵션
패딩 (List<Long> 패딩)

상속된 메서드

공개 방법

공개 출력 <T> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

public static XlaConcatND <T> create ( Scope 범위, Iterable< Operand <T>> 입력, List<Long> numConcats, Options... 옵션)

새로운 XlaConcatND 작업을 래핑하는 클래스를 생성하는 팩터리 메서드입니다.

매개변수
범위 현재 범위
입력 모든 차원에 걸쳐 병합하려면 행 우선 순서로 텐서 슬라이스를 입력하세요. 모든 입력은 동일한 형태를 가져야 합니다. } out_arg { 이름: "출력" 설명: <
numConcats 차원당 병합하는 방법의 수입니다.
옵션 선택적 속성 값을 전달합니다.
보고
  • XlaConcatND의 새 인스턴스

공개 출력 <T> 출력 ()

공개 정적 XlaConcatND.Options 패딩 (List<Long> 패딩)

매개변수
패딩 최종 병합된 텐서에서 제거할 차원당 오른쪽 패딩의 선택적 목록입니다. 이러한 패딩은 패딩을 제거하기 전에 병합된 결과의 차원 크기를 초과해서는 안 됩니다.