텐서플로:: 작전:: 모으다
#include <array_ops.h>
indices
에 따라 params
에서 조각을 수집 합니다.
요약
indices
는 모든 차원(일반적으로 0차원 또는 1차원)의 정수 텐서여야 합니다. 모양 indices.shape + params.shape[1:]
인 출력 텐서를 생성합니다. 여기서:
# Scalar indices output[:, ..., :] = params[indices, :, ... :]
# Vector indices output[i, :, ..., :] = params[indices[i], :, ... :]
# Higher rank indices output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]
indices
가 순열이고 len(indices) == params.shape[0]
경우 이 작업은 그에 따라 params
를 순열합니다.
validate_indices
: 지원 중단됨. 이 작업이 CPU에 할당되면 indices
의 값이 항상 범위 내에 있는지 확인합니다. GPU에 할당된 경우 범위를 벗어난 인덱스로 인해 안전하지만 지정되지 않은 동작이 발생하며 여기에는 오류 발생이 포함될 수 있습니다.
인수:
- 범위: 범위 개체
보고:
-
Output
: 출력 텐서입니다.
생성자와 소멸자 | |
---|---|
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices) | |
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::Attrs & attrs) |
공개 속성 | |
---|---|
operation | |
output |
공공 기능 | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
공개 정적 함수 | |
---|---|
ValidateIndices (bool x) |
구조체 | |
---|---|
tensorflow:: ops:: 수집:: 속성 | Gather 에 대한 선택적 속성 설정자. |
공개 속성
작업
Operation operation
산출
::tensorflow::Output output
공공 기능
모으다
Gather( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices )
모으다
Gather( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, const Gather::Attrs & attrs )
마디
::tensorflow::Node * node() const
연산자::텐서플로우::입력
operator::tensorflow::Input() const
연산자::텐서플로우::출력
operator::tensorflow::Output() const
공개 정적 함수
ValidateIndices
Attrs ValidateIndices( bool x )