tensorflow :: opérations :: BatchToSpace
#include <array_ops.h>
BatchToSpace pour les tenseurs 4-D de type T.
Résumé
Il s'agit d'une ancienne version du BatchToSpaceND plus général.
Réorganise (permute) les données du lot en blocs de données spatiales, suivi d'un recadrage. Il s'agit de la transformation inverse de SpaceToBatch. Plus spécifiquement, cette opération produit une copie du tenseur d'entrée où les valeurs de la dimension du batch
sont déplacées en blocs spatiaux vers les dimensions de height
et de width
, suivies d'un recadrage le long des dimensions de height
et de width
.
Arguments:
- scope: un objet Scope
- entrée: tenseur 4-D avec forme
[batch*block_size*block_size, height_pad/block_size, width_pad/block_size, depth]
. Notez que la taille du lot du tenseur d'entrée doit être divisible parblock_size * block_size
. - cultures: tenseur 2D d'entiers non négatifs de forme
[2, 2]
. Il spécifie le nombre d'éléments à rogner à partir du résultat intermédiaire dans les dimensions spatiales comme suit:crops = [[crop_top, crop_bottom], [crop_left, crop_right]]
Retour:
-
Output
: 4-D avec forme[batch, height, width, depth]
, où:height = height_pad - crop_top - crop_bottom width = width_pad - crop_left - crop_right
L'attr block_size
doit être supérieur à un. Il indique la taille du bloc.
Quelques exemples:
(1) Pour l'entrée suivante de la forme [4, 1, 1, 1]
et block_size de 2:
[[[[1]]], [[[2]]], [[[3]]], [[[4]]]]
Le tenseur de sortie a la forme [1, 2, 2, 1]
et la valeur:
x = [[[[1], [2]], [[3], [4]]]]
(2) Pour l'entrée suivante de la forme [4, 1, 1, 3]
et block_size de 2:
[[[[1, 2, 3]]], [[[4, 5, 6]]], [[[7, 8, 9]]], [[[10, 11, 12]]]]
Le tenseur de sortie a la forme [1, 2, 2, 3]
et la valeur:
x = [[[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]]
(3) Pour l'entrée suivante de la forme [4, 2, 2, 1]
et block_size de 2:
x = [[[[1], [3]], [[9], [11]]], [[[2], [4]], [[10], [12]]], [[[5], [7]], [[13], [15]]], [[[6], [8]], [[14], [16]]]]
Le tenseur de sortie a la forme [1, 4, 4, 1]
et la valeur:
x = [[[[1], [2], [3], [4]], [[5], [6], [7], [8]], [[9], [10], [11], [12]], [[13], [14], [15], [16]]]]
(4) Pour l'entrée suivante de la forme [8, 1, 2, 1]
et block_size de 2:
x = [[[[1], [3]]], [[[9], [11]]], [[[2], [4]]], [[[10], [12]]], [[[5], [7]]], [[[13], [15]]], [[[6], [8]]], [[[14], [16]]]]
Le tenseur de sortie a la forme [2, 2, 4, 1]
et la valeur:
x = [[[[1], [3]], [[5], [7]]], [[[2], [4]], [[10], [12]]], [[[5], [7]], [[13], [15]]], [[[6], [8]], [[14], [16]]]]
Constructeurs et destructeurs | |
---|---|
BatchToSpace (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input crops, int64 block_size) |
Attributs publics | |
---|---|
operation | |
output |
Fonctions publiques | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Attributs publics
opération
Operation operation
production
::tensorflow::Output output
Fonctions publiques
BatchToSpace
BatchToSpace( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input crops, int64 block_size )
nœud
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Output
operator::tensorflow::Output() const