`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> | 출력 () `params.shape[:axis] + indices.shape + params.shape[axis + 1:]` 형태로 `indices`에 의해 제공된 인덱스에서 수집된 `params`의 값입니다. |
상속된 메서드
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
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`의 값입니다.