텐서플로우:: 작전:: N번째요소

#include <nn_ops.h>

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

요약

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

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

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

인수:

  • 범위: 범위 개체
  • 입력: 1차원 이상, 마지막 차원이 n+1 이상입니다.
  • 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)

구조체

텐서플로우:: ops:: NthElement:: Attrs

NthElement 에 대한 선택적 속성 설정자입니다.

공개 속성

작업

Operation operation

가치

::tensorflow::Output values

공공 기능

N번째요소

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

N번째요소

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

마디

::tensorflow::Node * node() const 

연산자::텐서플로우::입력

 operator::tensorflow::Input() const 

연산자::텐서플로우::출력

 operator::tensorflow::Output() const 

공개 정적 함수

뒤집다

Attrs Reverse(
  bool x
)