Divisez un `SparseTensor` en tenseurs `num_split` le long d'une dimension.
Si le `shape[split_dim]` n'est pas un multiple entier de `num_split`. Les tranches `[0 : shape[split_dim] % num_split]` obtiennent une dimension supplémentaire. Par exemple, si `split_dim = 1` et `num_split = 2` et que l'entrée est
input_tensor = shape = [2, 7] [ ade ] [bc ]
Graphiquement, les tenseurs de sortie sont :
output_tensor[0] = shape = [2, 4] [ a ] [bc ]
output_tensor[1] = shape = [2, 3] [ de ] [ ]
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
statique <T étend TType > SparseSplit <T> | |
Liste< Sortie < TInt64 >> | |
Liste< Sortie < TInt64 >> | forme de sortie () Une liste de tenseurs 1D représente la forme des tenseurs clairsemés en sortie. |
Liste< Sortie <T>> | valeurs de sortie () Une liste de tenseurs 1D représente les valeurs des tenseurs clairsemés en sortie. |
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
Méthodes publiques
public static SparseSplit <T> créer ( Portée de portée , Opérande < TInt64 > splitDim, Opérande < TInt64 > indices, Valeurs de l' Opérande <T>, Opérande < TInt64 > forme, Long numSplit)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseSplit.
Paramètres
portée | portée actuelle |
---|---|
diviserDim | 0-D. Dimension selon laquelle diviser. Doit être compris dans la plage « [0, rang (forme)) ». |
indices | Le tenseur 2D représente les indices du tenseur clairsemé. |
valeurs | Le tenseur 1-D représente les valeurs du tenseur clairsemé. |
forme | 1-D. le tenseur représente la forme du tenseur clairsemé. indices de sortie : une liste de tenseurs 1-D représente les indices des tenseurs clairsemés de sortie. |
numSplit | Le nombre de façons de diviser. |
Retour
- une nouvelle instance de SparseSplit
liste publique < Sortie < TInt64 >> outputShape ()
Une liste de tenseurs 1D représente la forme des tenseurs clairsemés en sortie.
liste publique < Sortie <T>> valeurs de sortie ()
Une liste de tenseurs 1D représente les valeurs des tenseurs clairsemés en sortie.