flux tensoriel : : opérations : : SparseSplit

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

Résumé

Si le shape[split_dim] n'est pas un multiple entier de num_split . Slices [0 : shape[split_dim] % num_split] obtient 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]
[    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 :

• scope : un objet Scope
• split_dim : 0-D. Dimension selon laquelle diviser. Doit être compris dans la plage [0, rank(shape)) .
• 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.
• num_split : le nombre de façons de diviser.

Retours :

• OutputList sortieindices_sortie
• OutputList output_values : une liste de tenseurs 1D représente les valeurs des tenseurs clairsemés en sortie.
• OutputList output_shape : une liste de tenseurs 1D représente la forme des tenseurs clairsemés en sortie.

Attributs publics

Operation operation

::tensorflow::OutputList output_indices

::tensorflow::OutputList output_shape

::tensorflow::OutputList output_values

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