텐서플로우:: 연산:: 수집

#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에 할당되면 범위를 벗어난 인덱스는 안전하지만 명시되지 않은 동작을 유발하며, 오류 발생 등이 포함될 수 있습니다.

논거:

보고:

생성자와 소멸자

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)

구조

텐서플로우:: 연산:: 수집:: 속성

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 

공개 정적 함수

인덱스 유효성 검사

Attrs ValidateIndices(
  bool x
)