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.

Sauf indication contraire, le contenu de cette page est régi par une licence Creative Commons Attribution 4.0, et les échantillons de code sont régis par une licence Apache 2.0. Pour en savoir plus, consultez les Règles du site Google Developers. Java est une marque déposée d'Oracle et/ou de ses sociétés affiliées.

Dernière mise à jour le 2023/12/01 (UTC).