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

tensorflow :: ops :: MaxPoolWithArgmax

#include <nn_ops.h>

입력에 대해 최대 풀링을 수행하고 최대 값과 인덱스를 모두 출력합니다.

요약

argmax 의 인덱스는 평면화되어 [b, y, x, c] 위치의 최대 값은 평면화 된 인덱스가됩니다. (y * width + x) * channels + c if include_batch_in_index is False; ((b * height + y) * width + x) * channels + c include_batch_in_index 가 True 인 경우 ((b * height + y) * width + x) * channels + c .

패딩이 포함되어 있고 수학적으로 정답이 외부에 있더라도 (음수 또는 너무 큼) 평면화하기 전에 반환 된 인덱스는 항상 [0, height) x [0, width) )에 있습니다. 이것은 버그이지만, 특히 평면화로 인해 안전한 하위 호환 방식으로 수정하기가 어렵습니다.

인수 :

  • 범위 : 범위 개체
  • 입력 : 모양 [batch, height, width, channels] 4-D. 풀오버 입력 .
  • ksize : 입력 텐서의 각 차원에 대한 창의 크기.
  • strides : 입력 텐서의 각 차원에 대한 슬라이딩 윈도우의 보폭입니다.
  • padding : 사용할 채우기 알고리즘의 유형입니다.

선택적 속성 ( Attrs 참조) :

  • include_batch_in_index : argmax 평면화 된 인덱스에 배치 차원을 포함할지 여부.

보고:

  • Output 출력 : 최대 풀링 된 출력 텐서.
  • Output argmax : 4-D. 각 출력에 대해 선택된 최대 값의 평면화 된 인덱스입니다.

생성자와 소멸자

MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding)
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs)

공용 속성

argmax
operation
output

공개 정적 함수

IncludeBatchInIndex (bool x)
Targmax (DataType x)

구조체

tensorflow :: ops :: MaxPoolWithArgmax :: Attrs

MaxPoolWithArgmax에 대한 선택적 속성 설정자.

공용 속성

argmax

::tensorflow::Output argmax

조작

Operation operation

산출

::tensorflow::Output output

공공 기능

MaxPoolWithArgmax

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

MaxPoolWithArgmax

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const MaxPoolWithArgmax::Attrs & attrs
)

공개 정적 함수

IncludeBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)