tensor akışı:: işlem:: Seyrek Bölünmüş
#include <sparse_ops.h>
Bir SparseTensor
bir boyut boyunca num_split
tensörlere bölün.
Özet
Eğer shape[split_dim]
num_split
tamsayı katı değilse. Dilimler [0 : shape[split_dim] % num_split]
ekstra bir boyut alır. Örneğin, split_dim = 1
ve num_split = 2
ise ve giriş
input_tensor = shape = [2, 7] [ a d e ] [b c ]
Grafiksel olarak çıkış tensörleri şunlardır:
output_tensor[0] = shape = [2, 4] [ a ] [b c ] output_tensor[1] = shape = [2, 3] [ d e ] [ ]
Argümanlar:
- kapsam: Bir Kapsam nesnesi
- split_dim: 0-D. Bölünecek boyut.
[0, rank(shape))
aralığında olmalıdır. - endeksler: 2 boyutlu tensör, seyrek tensörün endekslerini temsil eder.
- değerler: 1-D tensör, seyrek tensörün değerlerini temsil eder.
- şekil: 1-D. tensör seyrek tensörün şeklini temsil eder. çıkış endeksleri: 1 boyutlu tensörlerin bir listesi, çıkış seyrek tensörlerinin endekslerini temsil eder.
- num_split: Bölme yollarının sayısı.
İade:
-
OutputList
çıktı_indisleri -
OutputList
çıktı_değerleri: 1 boyutlu tensörlerin bir listesi, çıkış seyrek tensörlerinin değerlerini temsil eder. -
OutputList
çıktı_şekli: 1 boyutlu tensörlerin bir listesi, çıktı seyrek tensörlerinin şeklini temsil eder.
Yapıcılar ve Yıkıcılar | |
---|---|
SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split) |
Genel özellikler | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
Genel özellikler
operasyon
Operation operation
çıktı_endeksleri
::tensorflow::OutputList output_indices
çıktı_şekli
::tensorflow::OutputList output_shape
çıktı_değerleri
::tensorflow::OutputList output_values
Kamu işlevleri
Seyrek Bölünmüş
SparseSplit( const ::tensorflow::Scope & scope, ::tensorflow::Input split_dim, ::tensorflow::Input indices, ::tensorflow::Input values, ::tensorflow::Input shape, int64 num_split )