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üyle:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[1, 1]
[[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>> | yineleyici () |
Liste< Çıkış <T>> | çıktılar () |
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çenekler)
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 |
İade
- XlaSplitND'nin yeni bir örneği
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. |
---|