Esta página foi traduzida pela API Cloud Translation.
Switch to English

tf.math.argmin

TensorFlow 1 versão Ver fonte no GitHub

Retorna o índice com o menor valor nos eixos de um tensor.

Retorna o menor índice em caso de empate.

input A Tensor . Deve ser um dos seguintes tipos: float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 .
axis A Tensor . Deve ser um dos seguintes tipos: int32 , int64 . int32 ou int64, deve estar no intervalo -rank(input), rank(input)) . Descreve qual o eixo do tensor de entrada para reduzir transversalmente. Para vectores, uso eixo = 0.
output_type Um opcional tf.DType de: tf.int32, tf.int64 . O padrão é tf.int64 .
name Um nome para a operação (opcional).

Um Tensor de tipo output_type .

Uso:

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