공개 최종 클래스 AssignVariableXlaConcatND
모든 차원에 걸쳐 입력 텐서를 연결합니다.
병합하는 연산은 주어진 num_splits 속성을 기반으로 입력 텐서를 분할하고, 선택적으로 패딩을 제거하고, 패딩 없이 병합된 텐서를 리소스 변수에 씁니다.
이 작업은 TPU 브리지를 통해 생성될 수 있습니다.
예를 들어 `입력` 텐서를 사용하면 다음과 같습니다.
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
`num_splits`: [2, 2]
그리고 `패딩`: [1, 1]
예상되는 '출력'은 다음과 같습니다. [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
중첩 클래스
수업 | AssignVariableXlaConcatND.Options | AssignVariableXlaConcatND 의 선택적 속성 |
공개 방법
정적 <T> AssignVariableXlaConcatND | 생성 ( Scope 범위, Operand <?> 리소스, Iterable< Operand <T>> 입력, List<Long> numConcats, Options... 옵션) 새로운 AssignVariableXlaConcatND 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다. |
정적 AssignVariableXlaConcatND.Options | 패딩 (List<Long> 패딩) |
상속된 메서드
공개 방법
공개 정적 AssignVariableXlaConcatND 생성 ( 범위 범위, 피연산자 <?> 리소스, Iterable< 피연산자 <T>> 입력, List<Long> numConcats, 옵션... 옵션)
새로운 AssignVariableXlaConcatND 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
의지 | 모든 차원에 걸쳐 연결된 입력 텐서를 위한 리소스 변수입니다. } in_arg { 이름: "입력" 설명: < |
numConcats | 차원당 병합하는 방법의 수입니다. |
옵션 | 선택적 속성 값을 전달합니다. |
보고
- AssignVariableXlaConcatND의 새 인스턴스
public static AssignVariableXlaConcatND.Options 패딩 (List<Long> 패딩)
매개변수
패딩 | 최종 병합된 텐서에서 제거할 차원당 오른쪽 패딩의 선택적 목록입니다. 이러한 패딩은 패딩을 제거하기 전에 병합된 결과의 차원 크기를 초과해서는 안 됩니다. |
---|