Aide à protéger la Grande barrière de corail avec tensorflow sur Kaggle Rejoignez Défi

tensorflow :: opérations :: OneHot

#include <array_ops.h>

Renvoie un tenseur one-hot.

Résumé

Les emplacements représentés par des indices dans les indices prennent la valeur on_value , tandis que tous les autres emplacements prennent la valeur off_value .

Si les indices entrée sont de rang N , la sortie aura le rang N+1 , Le nouvel axe est créé au niveau de l' axis dimension (par défaut: le nouvel axe est ajouté à la fin).

Si les indices sont un scalaire, la forme de sortie sera un vecteur de depth de longueur.

Si les indices sont un vecteur d' features de longueur, la forme en sortie sera:

  features x depth if axis == -1
  depth x features if axis == 0

Si les indices est une matrice (lot) avec une forme [batch, features] , la forme de sortie sera:

  batch x features x depth if axis == -1
  batch x depth x features if axis == 1
  depth x batch x features if axis == 0

Exemples

Supposons que

  indices = [0, 2, -1, 1]
  depth = 3
  on_value = 5.0
  off_value = 0.0
  axis = -1

La sortie est alors [4 x 3] :

output =
  [5.0 0.0 0.0]  // one_hot(0)
  [0.0 0.0 5.0]  // one_hot(2)
  [0.0 0.0 0.0]  // one_hot(-1)
  [0.0 5.0 0.0]  // one_hot(1)

Supposons que

  indices = [0, 2, -1, 1]
  depth = 3
  on_value = 0.0
  off_value = 3.0
  axis = 0

La sortie est alors [3 x 4] :

output =
  [0.0 3.0 3.0 3.0]
  [3.0 3.0 3.0 0.0]
  [3.0 3.0 3.0 3.0]
  [3.0 0.0 3.0 3.0]
//  ^                one_hot(0)
//      ^            one_hot(2)
//          ^        one_hot(-1)
//              ^    one_hot(1)

Supposons que

  indices = [[0, 2], [1, -1]]
  depth = 3
  on_value = 1.0
  off_value = 0.0
  axis = -1

La sortie est alors [2 x 2 x 3] :

output =
  [
    [1.0, 0.0, 0.0]  // one_hot(0)
    [0.0, 0.0, 1.0]  // one_hot(2)
  ][
    [0.0, 1.0, 0.0]  // one_hot(1)
    [0.0, 0.0, 0.0]  // one_hot(-1)
  ]

Arguments:

  • scope: un objet Scope
  • indices: un tenseur d'indices.
  • profondeur: un scalaire définissant la profondeur de la seule dimension chaude.
  • on_value: Un scalaire définissant la valeur à remplir en sortie lorsque indices[j] = i .
  • off_value: Un scalaire définissant la valeur à remplir en sortie lorsque indices[j] != i .

Attributs facultatifs (voir Attrs ):

  • axis: L'axe à remplir (par défaut: -1, un nouvel axe le plus à l'intérieur).

Retour:

  • Output : le tenseur one-hot.

Constructeurs et destructeurs

OneHot (const :: tensorflow::Scope & scope, :: tensorflow::Input indices, :: tensorflow::Input depth, :: tensorflow::Input on_value, :: tensorflow::Input off_value)
OneHot (const :: tensorflow::Scope & scope, :: tensorflow::Input indices, :: tensorflow::Input depth, :: tensorflow::Input on_value, :: tensorflow::Input off_value, const OneHot::Attrs & attrs)

Attributs publics

operation
output

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fonctions statiques publiques

Axis (int64 x)

Structs

tensorflow :: ops :: OneHot :: Attrs

Définisseurs d' attributs facultatifs pour OneHot .

Attributs publics

opération

Operation operation

production

::tensorflow::Output output

Fonctions publiques

OneHot

 OneHot(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input indices,
  ::tensorflow::Input depth,
  ::tensorflow::Input on_value,
  ::tensorflow::Input off_value
)

OneHot

 OneHot(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input indices,
  ::tensorflow::Input depth,
  ::tensorflow::Input on_value,
  ::tensorflow::Input off_value,
  const OneHot::Attrs & attrs
)

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const 

Fonctions statiques publiques

Axe

Attrs Axis(
  int64 x
)