tensorflow:: operaciones:: DispersoDividir

#include <sparse_ops.h>

Divide un SparseTensor en num_split tensores a lo largo de una dimensión.

Resumen

Si la shape[split_dim] no es un múltiplo entero de num_split . Las rebanadas [0 : shape[split_dim] % num_split] obtienen una dimensión extra. Por ejemplo, si split_dim = 1 y num_split = 2 y la entrada es

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

Gráficamente los tensores de salida son:

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

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

Argumentos:

  • alcance: un objeto de alcance
  • split_dim: 0-D. La dimensión a lo largo de la cual dividir. Debe estar en el rango [0, rank(shape)) .
  • índices: el tensor 2-D representa los índices del tensor disperso.
  • valores: el tensor 1-D representa los valores del tensor disperso.
  • forma: 1-D. tensor representa la forma del tensor disperso. índices de salida: una lista de tensores 1-D representa los índices de los tensores dispersos de salida.
  • num_split: El número de formas de dividir.

Devoluciones:

  • OutputList salida_índices
  • OutputList output_values: una lista de tensores 1-D representa los valores de los tensores dispersos de salida.
  • OutputList output_shape: una lista de tensores 1-D representa la forma de los tensores dispersos de salida.

Constructores y Destructores

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

Atributos públicos

operation
output_indices
output_shape
output_values

Atributos públicos

operación

Operation operation

índices_de_salida

::tensorflow::OutputList output_indices

salida_forma

::tensorflow::OutputList output_shape

valores_de_salida

::tensorflow::OutputList output_values

Funciones publicas

DispersoDividir

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