ReadVariableXlaSplitND

genel final sınıfı ReadVariableXlaSplitND

Kaynak değişkeni giriş tensörünü tüm boyutlara böler.

Verilen num_splits niteliğine göre kaynak değişkeni giriş tensörünü bölen, 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 ReadVariableXlaSplitND.Options ReadVariableXlaSplitND için isteğe bağlı özellikler

Genel Yöntemler

statik <T> ReadVariableXlaSplitND <T>
oluştur ( Kapsam kapsamı, İşlenen <?> kaynağı, Sınıf<T> T, Uzun N, Liste<Uzun> numSplits, Seçenekler... seçenekler)
Yeni bir ReadVariableXlaSplitND işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Yineleyici< İşlenen <T>>
Liste< Çıkış <T>>
statik ReadVariableXlaSplitND.Options
dolgular (<Uzun> dolguları listele)

Kalıtsal Yöntemler

Genel Yöntemler

public static ReadVariableXlaSplitND <T> create ( Kapsam kapsamı, İşlenen <?> kaynağı, Class<T> T, Long N, List<Long> numSplits, Options... seçenekleri)

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

Parametreler
kapsam mevcut kapsam
kaynak Tüm boyutlara bölünecek giriş tensörünün kaynak değişkeni. } 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
  • ReadVariableXlaSplitND'nin yeni bir örneği

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

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

public static ReadVariableXlaSplitND.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.