SpaceToBatch

classe finale publique SpaceToBatch

SpaceToBatch pour les tenseurs 4-D de type T.

Il s'agit d'une version héritée de SpaceToBatchND, plus général.

Ajoutez des zéros, puis réorganisez (permutez) les blocs de données spatiales en lots. Plus précisément, cette opération génère une copie du tenseur d'entrée où les valeurs des dimensions « hauteur » et « largeur » sont déplacées vers la dimension « lot ». Après le remplissage de zéros, la « hauteur » et la « largeur » de l'entrée doivent être divisibles par la taille du bloc.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique du tenseur.
statique <T étend TType > SpaceToBatch <T>
créer ( Scope scope, Operand <T> input, Operand <? extends TNumber > paddings, Long blockSize)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SpaceToBatch.
Sortie <T>
sortir ()

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "SpaceToBatch"

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique du tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static SpaceToBatch <T> create ( Scope scope, Operand <T> input, Operand <? extends TNumber > paddings, Long blockSize)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération SpaceToBatch.

Paramètres
portée portée actuelle
saisir 4-D avec la forme `[lot, hauteur, largeur, profondeur]`.
rembourrages Tenseur 2D d'entiers non négatifs de forme `[2, 2]`. Il spécifie le remplissage de l'entrée avec des zéros dans les dimensions spatiales comme suit :

rembourrages = [[pad_top, pad_bottom], [pad_left, pad_right]]

Les dimensions spatiales effectives du tenseur d'entrée avec remplissage de zéros seront :

height_pad = pad_top + hauteur + pad_bottom width_pad = pad_left + largeur + pad_right

L'attr `block_size` doit être supérieur à un. Il indique la taille du bloc.

* Les blocs qui ne se chevauchent pas de taille « block_size x block size » dans les dimensions de hauteur et de largeur sont réorganisés dans la dimension du lot à chaque emplacement. * Le lot du tenseur de sortie est `batch * block_size * block_size`. * Height_pad et width_pad doivent être divisibles par block_size.

La forme du résultat sera :

[batch block_size block_size, height_pad/block_size, width_pad/block_size, profondeur]

Quelques exemples :

(1) Pour l'entrée suivante de forme `[1, 2, 2, 1]` et block_size de 2 : 

x = [[[[1], [2]], [[3], [4]]]]
Le tenseur de sortie a la forme `[4, 1, 1, 1]` et la valeur : 
[[[[1]]], [[[2]]], [[[3]]], [[[4]]]]
(2) Pour l'entrée suivante de forme `[1, 2, 2, 3]` et block_size de 2 : 
x = [[[[1, 2, 3], [4, 5, 6]],
       [[7, 8, 9], [10, 11, 12]]]]
Le tenseur de sortie a la forme `[4, 1, 1, 3]` et la valeur : 
[[[[1, 2, 3]]], [[[4, 5, 6]]], [[[7, 8, 9]]], [[[10, 11, 12]]]]
(3) Pour l'entrée suivante de forme `[1, 4, 4, 1]` et block_size de 2 : 
x = [[[[1],   [2],  [3],  [4]],
       [[5],   [6],  [7],  [8]],
       [[9],  [10], [11],  [12]],
       [[13], [14], [15],  [16]]]]
Le tenseur de sortie a la forme `[4, 2, 2, 1]` et la valeur : 
x = [[[[1], [3]], [[9], [11]]],
      [[[2], [4]], [[10], [12]]],
      [[[5], [7]], [[13], [15]]],
      [[[6], [8]], [[14], [16]]]]
(4) Pour l'entrée suivante de forme `[2, 2, 4, 1]` et block_size de 2 : 
x = [[[[1],   [2],  [3],  [4]],
       [[5],   [6],  [7],  [8]]],
      [[[9],  [10], [11],  [12]],
       [[13], [14], [15],  [16]]]]
Le tenseur de sortie a la forme `[8, 1, 2, 1]` et la valeur : 
x = [[[[1], [3]]], [[[9], [11]]], [[[2], [4]]], [[[10], [12]]],
      [[[5], [7]]], [[[13], [15]]], [[[6], [8]]], [[[14], [16]]]]
Entre autres, cette opération est utile pour réduire la convolution atreuse en convolution régulière.

Retours
  • une nouvelle instance de SpaceToBatch

sortie publique <T> sortie ()