tensorflow::ops::SparseSplit

#include <sparse_ops.h>

Diviser un SparseTensor en num_split tenseurs le long d'une dimension.

Résumé

If the shape[split_dim] is not an integer multiple of num_split . Slices [0 : shape[split_dim] % num_split] gets one extra dimension. For example, if split_dim = 1 and num_split = 2 and the input is

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

Graphiquement, les tenseurs de sortie sont :

output_tensor[0] = shape = [2, 4]
[    a  ]
[b c    ]

output_tensor[1] = shape = [2, 3]
[ d e  ]
[      ]

Arguments :

  • portée : un objet Scope
  • split_dim : 0-D. La dimension selon laquelle diviser. Doit être dans la plage [0, rank(shape)) .
  • indices : le tenseur 2D représente les indices du tenseur creux.
  • valeurs : le tenseur 1D représente les valeurs du tenseur creux.
  • forme : tenseur 1D représentant la forme du tenseur creux. indices de sortie : une liste de tenseurs 1D représentant les indices des tenseurs creux de sortie.
  • num_split : Le nombre de façons de diviser.

Retours :

  • OutputList indices_de_sortie
  • OutputList output_values : Une liste de tenseurs 1D représentant les valeurs des tenseurs creux de sortie.
  • OutputList output_shape : Une liste de tenseurs 1D représente la forme des tenseurs creux de sortie.

Constructeurs et destructeurs

SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split)

attributs publics

operation
output_indices
output_shape
output_values

attributs publics

opération 

Operation operation

indices_de_sortie 

::tensorflow::OutputList output_indices

forme_de_sortie 

::tensorflow::OutputList output_shape

valeurs_de_sortie 

::tensorflow::OutputList output_values

fonctions publiques

SparseSplit 

 SparseSplit(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input split_dim,
  ::tensorflow::Input indices,
  ::tensorflow::Input values,
  ::tensorflow::Input shape,
  int64 num_split
)