тензорный поток:: опс:: АргМин
#include <math_ops.h>
Возвращает индекс с наименьшим значением по измерениям тензора.
Краткое содержание
Обратите внимание, что в случае связей идентичность возвращаемого значения не гарантируется.
Использование:
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
Аргументы:
- область: объект области.
- размерность: int32 или int64, должна находиться в диапазоне
[-rank(input), rank(input))
. Описывает, какое измерение входного тензора нужно уменьшить. Для векторов используйте размерность = 0.
Возврат:
-
Output
: выходной тензор.
Конструкторы и деструкторы | |
---|---|
ArgMin (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input dimension) | |
ArgMin (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input dimension, const ArgMin::Attrs & attrs) |
Публичные атрибуты | |
---|---|
operation | |
output |
Общественные функции | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Публичные статические функции | |
---|---|
OutputType (DataType x) |
Структуры | |
---|---|
tensorflow::ops::ArgMin::Attrs | Необязательные установщики атрибутов для ArgMin . |
Публичные атрибуты
операция
Operation operation
выход
::tensorflow::Output output
Общественные функции
АргМин
ArgMin( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input dimension )
АргМин
ArgMin( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input dimension, const ArgMin::Attrs & attrs )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
Тип вывода
Attrs OutputType( DataType x )