Diese Seite wurde von der Cloud Translation API übersetzt.
Switch to English

Tensorflow :: ops :: SparseSplit

#include <sparse_ops.h>

SparseTensor num_split einen SparseTensor entlang einer Dimension in num_split Tensoren auf.

Zusammenfassung

Wenn die shape[split_dim] kein ganzzahliges Vielfaches von num_split . Slices [0 : shape[split_dim] % num_split] eine zusätzliche Dimension. Zum Beispiel, wenn split_dim = 1 und num_split = 2 und die Eingabe

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

Grafisch sind die Ausgangstensoren:

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

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

Argumente:

  • scope: Ein Scope- Objekt
  • split_dim: 0-D. Die Dimension, entlang der geteilt werden soll. Muss im Bereich [0, rank(shape)) .
  • Indizes: Der 2D-Tensor repräsentiert die Indizes des spärlichen Tensors.
  • Werte: 1-D-Tensor repräsentiert die Werte des spärlichen Tensors.
  • Form: 1-D. Tensor repräsentiert die Form des spärlichen Tensors. Ausgangsindizes: Eine Liste von 1-D-Tensoren repräsentiert die Indizes der Ausgangstensoren.
  • num_split: Die Anzahl der Aufteilungsmöglichkeiten.

Kehrt zurück:

  • OutputList output_indices
  • OutputList output_values: Eine Liste von 1-D-Tensoren repräsentiert die Werte der Ausgabe-Tensoren mit OutputList Dichte.
  • OutputList output_shape: Eine Liste von 1-D-Tensoren repräsentiert die Form der Tensoren mit OutputList Dichte.

Konstruktoren und Destruktoren

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

Öffentliche Attribute

operation
output_indices
output_shape
output_values

Öffentliche Attribute

Operation

Operation operation

output_indices

::tensorflow::OutputList output_indices

output_shape

::tensorflow::OutputList output_shape

Ausgabewerte

::tensorflow::OutputList output_values

Öffentliche Funktionen

SparseSplit

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