Gather

공개 최종 수업 모여라

문서화된 XLA Gather 연산자를 래핑합니다.

https://www.tensorflow.org/xla/Operation_semantics#gather

상수

OP_NAME TensorFlow 코어 엔진에서 알려진 이 작업의 이름

공개 방법

출력 <T>
출력 ()
텐서의 기호 핸들을 반환합니다.
static <T는 TType을 확장하고, U는 TNumber를 확장하고 > Gather <T>를 확장합니다.
생성 ( 범위 범위, 피연산자 <T> 피연산자, 피연산자 <U> startIndices, 피연산자 <U> SliceSizes, 문자열 차원Numbers, 부울 인덱스AreSorted)
새로운 Gather 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <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의 새로운 인스턴스

공개 출력 <T> 출력 ()