공개 최종 클래스 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> | |
정적 ArgMax < TInt64 > | |
출력 <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의 새 인스턴스