flux tensoriel : : opérations : : Diffusion vers
#include <array_ops.h>
Diffusez un tableau pour une forme compatible.
Résumé
La diffusion est le processus de création de tableaux ayant des formes compatibles pour les opérations arithmétiques. Deux formes sont compatibles si pour chaque paire de dimensions elles sont égales ou si l'une d'elles en est une. Lorsque vous essayez de diffuser un Tensor sur une forme, il commence par les dimensions de fin et progresse.
Par exemple,
x = tf.constant([1, 2, 3]) y = tf.broadcast_to(x, [3, 3]) print(y) tf.Tensor( [[1 2 3] [1 2 3] [1 2 3]], forme=(3, 3), dtype=int32)
Dans l'exemple ci-dessus, le Tensor d'entrée avec la forme de [1, 3]
est diffusé vers le Tensor de sortie avec la forme de [3, 3]
.
Lors d'opérations de diffusion telles que la multiplication d'un tenseur par un scalaire, la diffusion confère (généralement) un certain avantage temporel ou spatial, car le tenseur diffusé n'est jamais matérialisé.
Cependant, broadcast_to
n’apporte pas de tels avantages. Le tenseur nouvellement créé prend toute la mémoire de la forme diffusée. (Dans un contexte graphique, broadcast_to
peut cependant être fusionné avec une opération ultérieure, puis optimisé.)
Arguments :
- scope : un objet Scope
- entrée : Un Tensor à diffuser.
- shape : un tenseur
int
1-D. La forme de la sortie souhaitée.
Retours :
Constructeurs et Destructeurs | |
---|---|
BroadcastTo (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input 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
Diffusion vers
BroadcastTo( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input shape )
nœud
::tensorflow::Node * node() const
opérateur :: tensorflow :: Entrée
operator::tensorflow::Input() const
opérateur :: tensorflow :: Sortie
operator::tensorflow::Output() const