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

tensorflow :: ops :: NthElement

#include <nn_ops.h>

마지막 차원에 대한 n 차 통계 값을 찾습니다.

요약

입력 값이 벡터 (순위 -1)이면 벡터에서 n 번째로 작은 항목을 찾아 해당 값을 스칼라 텐서로 출력합니다.

행렬의 경우 (상위 순위 입력) 각 행에서 n 번째로 작은 값 (각각 마지막 차원을 따라 벡터) 인 항목을 계산합니다. 따라서

values.shape = input.shape[:-1]

인수 :

  • 범위 : 범위 개체
  • 입력 : 마지막 차원이 n+1 이상인 1-D 이상.
  • n : 0-D. 행렬의 각 행을 따라 마지막 차원을 따라 선택할 정렬 된 벡터의 위치입니다. n의 유효한 범위는 [0, input.shape[:-1])

선택적 속성 ( Attrs 참조) :

  • reverse : True로 설정하면 벡터에서 n 번째로 큰 값을 찾고 그 반대의 경우도 마찬가지입니다.

보고:

  • Output : 각 마지막 차원 조각을 따라 n 차 통계.

생성자와 소멸자

NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n)
NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n, const NthElement::Attrs & attrs)

공용 속성

operation
values

공공 기능

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

공개 정적 함수

Reverse (bool x)

구조체

tensorflow :: ops :: NthElement :: Attrs

NthElement에 대한 선택적 속성 setter입니다.

공용 속성

조작

Operation operation

가치

::tensorflow::Output values

공공 기능

NthElement

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n
)

NthElement

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n,
  const NthElement::Attrs & attrs
)

마디

::tensorflow::Node * node() const 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const 

공개 정적 함수

역전

Attrs Reverse(
  bool x
)