tensorflow :: ops :: SparseSplit

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

Podsumowanie

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

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

Graficznie tensory wyjściowe to:

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

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

Argumenty:

• zakres: obiekt Scope
• split_dim: 0-D. Wymiar, wzdłuż którego ma nastąpić 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 rzadkiego tensora. wskaźniki wyjściowe: lista tensorów 1-D reprezentuje indeksy wyjściowych tensorów rzadkich.
• num_split: liczba sposobów podziału.

Zwroty:

• OutputList output_indices
• OutputList output_values: lista tensorów 1-D reprezentuje wartości wyjściowych tensorów rzadkich.
• OutputList output_shape: lista tensorów 1-D reprezentuje kształt wyjściowych tensorów rzadkich.

Atrybuty publiczne

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
)