Google I/O에 참여해 주셔서 감사합니다. 주문형 모든 세션 보기 주문형 시청

텐서플로:: 작전:: 모으다

#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
)