tensorflow:: ops:: SparseSplit

#include <sparse_ops.h>

Podziel SparseTensor na tensory num_split wzdłuż jednego wymiaru.

Streszczenie

Jeśli shape[split_dim] nie jest całkowitą wielokrotnością num_split . Wycinki [0 : shape[split_dim] % num_split] otrzymują jeden dodatkowy wymiar. Na przykład, jeśli split_dim = 1 i num_split = 2 , a dane wejściowe to

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

Graficznie tensory wyjściowe przedstawiają się następująco:

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

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

Argumenty:

  • zakres: obiekt zakresu
  • split_dim: 0-D. Wymiar, wzdłuż którego następuje podział. Musi mieścić się w zakresie [0, rank(shape)) .
  • indeksy: tensor 2-D reprezentuje indeksy tensora rzadkiego.
  • wartości: Tensor 1-D reprezentuje wartości tensora rzadkiego.
  • kształt: 1-D. Tensor reprezentuje kształt tensora rzadkiego. Indeksy wyjściowe: Lista tensorów 1-D reprezentuje indeksy wyjściowych tensorów rzadkich.
  • num_split: Liczba sposobów podziału.

Zwroty:

  • OutputList indeksy_wyjściowe
  • OutputList output_values: Lista tensorów 1-D reprezentująca wartości wyjściowych tensorów rzadkich.
  • OutputList output_shape: Lista tensorów 1-D reprezentująca kształt wyjściowych tensorów rzadkich.

Konstruktory i destruktory

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

Atrybuty publiczne

operation
output_indices
output_shape
output_values

Atrybuty publiczne

działanie 

Operation operation

indeksy_wyjściowe 

::tensorflow::OutputList output_indices

kształt_wyjściowy 

::tensorflow::OutputList output_shape

wartości_wyjściowe 

::tensorflow::OutputList output_values

Funkcje publiczne

Rozrzedzony podział 

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