flux tensoriel : : opérations : : MaxPoolAvecArgmax :: Attributs

#include <nn_ops.h>

Setters d'attributs facultatifs pour MaxPoolWithArgmax .

Résumé

Attributs publics

Targmax_ = DT_INT64
DataType
include_batch_in_index_ = false
bool

Fonctions publiques

IncludeBatchInIndex (bool x)
TF_MUST_USE_RESULT Attrs
S'il faut inclure la dimension du lot dans l'index aplati de argmax .
Targmax (DataType x)
TF_MUST_USE_RESULT Attrs
La valeur par défaut est DT_INT64.

Attributs publics

Targmax_

DataType tensorflow::ops::MaxPoolWithArgmax::Attrs::Targmax_ = DT_INT64

include_batch_in_index_

bool tensorflow::ops::MaxPoolWithArgmax::Attrs::include_batch_in_index_ = false

Fonctions publiques

InclureBatchInIndex

TF_MUST_USE_RESULT Attrs tensorflow::ops::MaxPoolWithArgmax::Attrs::IncludeBatchInIndex(
  bool x
)

S'il faut inclure la dimension du lot dans l'index aplati de argmax .

La valeur par défaut est false

Targmax

TF_MUST_USE_RESULT Attrs tensorflow::ops::MaxPoolWithArgmax::Attrs::Targmax(
  DataType x
)

La valeur par défaut est DT_INT64.