flux tensoriel : : opérations : : LotVersEspace

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]]]]

Attributs publics

Operation operation

::tensorflow::Output output

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const