flux tensoriel : : opérations : : LotVersEspace

#include <array_ops.h>

BatchToSpace pour les tenseurs 4-D de type T.

Résumé

Il s'agit d'une version héritée 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 précisément, cette opération génère une copie du tenseur d'entrée où les valeurs de la dimension batch sont déplacées dans des blocs spatiaux vers les dimensions height et width , suivies d'un recadrage le long des dimensions height et width .

Arguments :

  • scope : un objet Scope
  • entrée : tenseur 4D 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 par block_size * block_size .
  • cultures : tenseur 2D d'entiers non négatifs de forme [2, 2] . Il spécifie le nombre d'éléments à recadrer à partir du résultat intermédiaire dans les dimensions spatiales, comme suit :
    crops = [[crop_top, crop_bottom], [crop_left, crop_right]]
    

Retours :

  • 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 shape [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 shape [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 shape [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 shape [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

sortir

::tensorflow::Output output

Fonctions publiques

LotVersEspace

 BatchToSpace(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input crops,
  int64 block_size
)

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: Entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const