수집

공개 최종 수업 모여라

`indices`에 따라 `params` 축 `axis`에서 슬라이스를 수집합니다.

`indices`는 모든 차원(보통 0-D 또는 1-D)의 정수 텐서여야 합니다. `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` 형태의 출력 텐서를 생성합니다. 여기서:

# Scalar indices (output is rank(params) - 1).
     output[a_0, ..., a_n, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices, b_0, ..., b_n]
 
     # Vector indices (output is rank(params)).
     output[a_0, ..., a_n, i, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
 
     # Higher rank indices (output is rank(params) + rank(indices) - 1).
     output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
       params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
 

CPU에서 범위를 벗어난 인덱스가 발견되면 오류가 반환됩니다. GPU에서는 범위를 벗어난 인덱스가 발견되면 해당 출력 값에 0이 저장됩니다.

`tf.batch_gather` 및 `tf.gather_nd`도 참조하세요.

중첩 클래스

수업 수집.옵션 Gather 의 선택적 속성

공개 방법

출력 <T>
출력 ()
텐서의 기호 핸들을 반환합니다.
정적 Gather.옵션
BatchDims (긴 BatchDims)
static <T, U 확장 번호, V 확장 번호> 수집 <T>
생성 ( 범위 범위, 피연산자 <T> 매개변수, 피연산자 <U> 인덱스, 피연산자 <V> 축, 옵션... 옵션)
새로운 Gather 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <T>
출력 ()
`params.shape[:axis] + indices.shape + params.shape[axis + 1:]` 형태로 `indices`에 의해 제공된 인덱스에서 수집된 `params`의 값입니다.

상속된 메서드

공개 방법

공개 출력 <T> asOutput ()

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

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

공개 정적 Gather.Options 배치Dims (Long 배치Dims)

public static Gather <T> create ( 범위 범위, 피연산자 <T> 매개 변수, 피연산자 <U> 인덱스, 피연산자 <V> 축, 옵션... 옵션)

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

매개변수
범위 현재 범위
매개변수 값을 수집할 텐서입니다. 최소한 '축 + 1' 등급이어야 합니다.
지수 인덱스 텐서. `[0, params.shape[axis])` 범위 내에 있어야 합니다.
중심선 '인덱스'를 수집할 'params'의 축입니다. 기본값은 첫 번째 차원입니다. 음수 인덱스를 지원합니다.
옵션 선택적 속성 값을 전달합니다.
보고
  • Gather의 새로운 인스턴스

공개 출력 <T> 출력 ()

`params.shape[:axis] + indices.shape + params.shape[axis + 1:]` 형태로 `indices`에 의해 제공된 인덱스에서 수집된 `params`의 값입니다.