flux tensoriel : : opérations : : Assurer la forme

#include <array_ops.h>

Garantit que la forme du tenseur correspond à la forme attendue.

Résumé

Génère une erreur si la forme du tenseur d'entrée ne correspond pas à la forme spécifiée. Renvoie le tenseur d'entrée sinon.

Arguments :

  • scope : un objet Scope
  • entrée : Un tenseur, dont la forme doit être validée.
  • forme : la forme attendue (éventuellement partiellement spécifiée) du tenseur d'entrée.

Retours :

  • Output : un tenseur avec la même forme et le même contenu que le tenseur ou la valeur d'entrée.

Constructeurs et Destructeurs

EnsureShape (const :: tensorflow::Scope & scope, :: tensorflow::Input input, PartialTensorShape shape)

Attributs publics

operation
output

Fonctions publiques

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

Attributs publics

opération

Operation operation

sortir

::tensorflow::Output output

Fonctions publiques

Assurer la forme

 EnsureShape(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  PartialTensorShape shape
)

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: Entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const