tensoreflusso:: ops:: SparseSplit

#include <sparse_ops.h>

Dividi uno SparseTensor in tensori num_split lungo una dimensione.

Riepilogo

Se shape[split_dim] non è un multiplo intero di num_split . Le fette [0 : shape[split_dim] % num_split] ottengono una dimensione extra. Ad esempio, se split_dim = 1 e num_split = 2 e l'input è 

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

Graficamente i tensori di uscita sono: 

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

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

Argomenti:

  • scope: un oggetto Scope
  • split_dim: 0-D. La dimensione lungo la quale dividere. Deve essere compreso nell'intervallo [0, rank(shape)) .
  • indici: il tensore 2-D rappresenta gli indici del tensore sparso.
  • valori: il tensore 1-D rappresenta i valori del tensore sparso.
  • forma: 1-D. il tensore rappresenta la forma del tensore sparso. indici di output: un elenco di tensori 1-D rappresenta gli indici dei tensori sparsi di output.
  • num_split: il numero di modi per dividere.

Resi:

  • OutputList output_indices
  • OutputList output_values: un elenco di tensori 1-D rappresenta i valori dei tensori sparsi di output.
  • OutputList output_shape: un elenco di tensori 1-D rappresenta la forma dei tensori sparsi di output.

Costruttori e distruttori

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

Attributi pubblici

operation
output_indices
output_shape
output_values

Attributi pubblici

operazione 

Operation operation

indici_output 

::tensorflow::OutputList output_indices

forma_output 

::tensorflow::OutputList output_shape

valori_di_output 

::tensorflow::OutputList output_values

Funzioni pubbliche

SparseSplit

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