Restez organisé à l'aide des collections Enregistrez et classez les contenus selon vos préférences.

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