ArgMax

공개 최종 클래스 ArgMax

텐서의 차원 전체에서 가장 큰 값을 가진 인덱스를 반환합니다.

동점인 경우 반환 값의 동일성은 보장되지 않습니다.

용법:

import tensorflow as tf
   a = [1, 10, 26.9, 2.8, 166.32, 62.3]
   b = tf.math.argmax(input = a)
   c = tf.keras.backend.eval(b)
   # c = 4
   # here a[4] = 166.32 which is the largest element of a across axis 0
   

상수

OP_NAME TensorFlow 코어 엔진에서 알려진 이 작업의 이름

공개 방법

출력 <V>
출력 ()
텐서의 기호 핸들을 반환합니다.
static <V는 TNumber를 확장합니다 > ArgMax <V>
생성 ( 범위 범위, 피연산자 <? 확장 TType > 입력, 피연산자 <? 확장 TNumber > 차원, Class<V> 출력 유형)
새로운 ArgMax 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
정적 ArgMax < TInt64 >
생성 (범위 범위 , 피연산자 <? 확장 TType > 입력, 피연산자 <? 확장 TNumber > 차원)
기본 출력 유형을 사용하여 새 ArgMax 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <V>
출력 ()

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

TensorFlow 코어 엔진에서 알려진 이 작업의 이름

상수 값: "ArgMax"

공개 방법

공개 출력 <V> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

public static ArgMax <V> create ( 범위 범위, 피연산자 <? 확장 TType > 입력, 피연산자 <? 확장 TNumber > 차원, Class<V> 출력 유형)

새로운 ArgMax 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
차원 int32 또는 int64는 '[-rank(input), Rank(input))' 범위에 있어야 합니다. 축소할 입력 텐서의 차원을 설명합니다. 벡터의 경우 차원 = 0을 사용합니다.
보고
  • ArgMax의 새 인스턴스

public static ArgMax < TInt64 > create ( 범위 범위, 피연산자 <? 확장 TType > 입력, 피연산자 <? 확장 TNumber > 차원)

기본 출력 유형을 사용하여 새 ArgMax 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
차원 int32 또는 int64는 '[-rank(input), Rank(input))' 범위에 있어야 합니다. 축소할 입력 텐서의 차원을 설명합니다. 벡터의 경우 차원 = 0을 사용합니다.
보고
  • ArgMax의 새 인스턴스

공개 출력 <V> 출력 ()