ML 커뮤니티 데이는 11월 9일입니다! TensorFlow, JAX에서 업데이트를 우리와 함께, 더 자세히 알아보기

tensorflow :: ops :: 모으다

#include <array_ops.h>

indices 에 따라 params 에서 조각을수집 합니다.

요약

indices 는 모든 차원의 정수 텐서 여야합니다 (일반적으로 0-D 또는 1-D). 모양 indices.shape + params.shape[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 : DEPRECATED. 이 작업이 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)

구조체

tensorflow :: ops :: Gather :: Attrs

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 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const 

공개 정적 함수

ValidateIndices

Attrs ValidateIndices(
  bool x
)