Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tensorflow :: ops :: SparseSplit

#include <sparse_ops.h>

Pisahkan SparseTensor menjadi tensor num_split dalam satu dimensi.

Ringkasan

Jika shape[split_dim] bukan merupakan kelipatan bilangan bulat dari num_split . Irisan [0 : shape[split_dim] % num_split] mendapat satu dimensi tambahan. Misalnya, jika split_dim = 1 dan num_split = 2 dan inputnya adalah

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

Secara grafis tensor keluarannya adalah:

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

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

Argumen:

  • scope: Objek Scope
  • split_dim: 0-D. Dimensi yang akan dipisahkan. Harus dalam kisaran [0, rank(shape)) .
  • indeks: tensor 2-D mewakili indeks dari tensor jarang.
  • nilai: tensor 1-D mewakili nilai-nilai tensor renggang.
  • Bentuk: 1-D. tensor mewakili bentuk tensor jarang. indeks keluaran: Daftar tensor 1-D mewakili indeks tensor jarang keluaran.
  • num_split: Jumlah cara untuk memisahkan.

Pengembalian:

  • OutputList output_indices
  • OutputList output_values: Daftar tensor 1-D mewakili nilai tensor jarang keluaran.
  • OutputList output_shape: Daftar tensor 1-D mewakili bentuk tensor jarang keluaran.

Pembuat dan Penghancur

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

Atribut publik

operation
output_indices
output_shape
output_values

Atribut publik

operasi

Operation operation

output_indices

::tensorflow::OutputList output_indices

bentuk_luaran

::tensorflow::OutputList output_shape

output_values

::tensorflow::OutputList output_values

Fungsi publik

SparseSplit

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