XlaSplitND

genel final dersi XlaSplitND

Giriş tensörünü tüm boyutlara böler.

Verilen num_splits niteliğine göre giriş tensörünü dilimleyen, isteğe bağlı olarak dilimleri dolduran ve dilimleri döndüren bir işlem. Dilimler ana satır sırasına göre döndürülür.

Bu işlem TPU köprüsü aracılığıyla oluşturulabilir.

Örneğin, "giriş" tensörü:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 
"num_splits":
[2, 2]
 
ve "dolgu":
[1, 1]
 
beklenen "çıkışlar" şöyledir:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

İç İçe Sınıflar

sınıf XlaSplitND.Seçenekler XlaSplitND için isteğe bağlı özellikler

Genel Yöntemler

statik <T> XlaSplitND <T>
oluştur ( Kapsam kapsamı, İşlenen <T> girişi, Uzun N, <Uzun> numSplitleri Listele, Seçenekler... seçenekleri)
Yeni bir XlaSplitND işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Yineleyici< İşlenen <T>>
Liste< Çıkış <T>>
statik XlaSplitND.Seçenekler
dolgular (<Uzun> dolguları listele)

Kalıtsal Yöntemler

Genel Yöntemler

public static XlaSplitND <T> create ( Kapsam kapsamı, İşlenen <T> girişi, Uzun N, <Uzun> numSplitleri Listele, Seçenekler... seçenekleri)

Yeni bir XlaSplitND işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Parametreler
kapsam mevcut kapsam
giriş Tüm boyutlara bölünecek tensörü girin. } out_arg { isim: "çıkışlar" açıklaması: <
numSplits Boyut başına bölme yollarının sayısı. Şekil boyutları eşit olarak bölünebilir olmalıdır.
seçenekler isteğe bağlı nitelik değerlerini taşır
İadeler
  • XlaSplitND'nin yeni bir örneği

public Yineleyici< İşlenen <T>> yineleyici ()

genel Liste< Çıkış <T>> çıkışlar ()

genel statik XlaSplitND.Options dolguları (List<Long> dolguları)

Parametreler
dolgular Bölmeden önce uygulanacak giriş tensörünün boyutu başına sağ dolguların isteğe bağlı listesi. Bu, bir boyutu eşit şekilde bölünebilir hale getirmek için kullanılabilir.