공개 최종 수업 모여라
문서화된 XLA Gather 연산자를 래핑합니다.
https://www.tensorflow.org/xla/Operation_semantics#gather
상수
끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
static <T는 TType을 확장하고, U는 TNumber를 확장하고 > Gather <T>를 확장합니다. | |
출력 <T> | 출력 () |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "XlaGather"
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
public static Gather <T> create ( 범위 범위, 피연산자 <T> 피연산자, 피연산자 <U> startIndices, 피연산자 <U> SliceSizes, 문자열 차원Numbers, 부울 인덱스AreSorted)
새로운 Gather 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
피연산자 | 우리가 수집할 배열입니다. |
시작 인덱스 | 우리가 수집하는 조각의 시작 인덱스를 포함하는 배열입니다. |
슬라이스 크기 | Slice_sizes[i]는 차원 i의 슬라이스 경계입니다. |
차원번호 | 직렬화된 xla::GatherDimensionNumbers proto입니다. |
인덱스가정렬됨 | 인덱스가 정렬되었는지 여부를 나타내는 부울입니다. |
보고
- Gather의 새로운 인스턴스