Giriş işleneninin min/maks k değerlerini ve bunların endekslerini yaklaşık olarak döndürür.
Algoritma ayrıntıları için https://arxiv.org/abs/2206.14286 adresine bakın. Bu işlem şu anda yalnızca TPU'da optimize edilmiştir.
İç İçe Sınıflar
sınıf | YaklaşıkÜstK.Seçenekler | ApproxTopK için isteğe bağlı özellikler |
Genel Yöntemler
statik YaklaşıkTopK.Seçenekler | agregaToTopk (Boolean agregaToTopk) |
statik <T Sayıyı genişletir> ApproxTopK <T> | oluştur ( Kapsam kapsamı, İşlenen <T> girişi, Uzun k, Seçenekler... seçenekler) Yeni bir ApproxTopK işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi. |
Çıkış <Tamsayı> | endeksler () "Giriş" işleneninin "reduction_dimension" boyunca "değerler" endeksleri. |
statik YaklaşıkTopK.Seçenekler | isMaxK (Boolean isMaxK) |
statik YaklaşıkTopK.Seçenekler | geri çağırmaTarget (Kayan geri çağırmaTarget) |
statik YaklaşıkTopK.Seçenekler | KüçültmeBoyut (Uzun KüçültmeBoyut) |
statik YaklaşıkTopK.Seçenekler | redüksiyonInputSizeOverride (Uzun redüksiyonInputSizeOverride) |
Çıkış <T> | değerler () "Giriş" işleneninin "redüksiyon_boyutu" boyunca min/maks k değerleri. |
Kalıtsal Yöntemler
Genel Yöntemler
genel statik ApproxTopK.Options agregaToTopk (Boolean agregaToTopk)
Parametreler
agregaToTopk | Doğru olduğunda, yaklaşık sonuçları en üst k'ya toplar. Yanlış olduğunda yaklaşık sonuçları döndürür. Yaklaşık sonuçların sayısı uygulama tarafından tanımlanır ve belirtilen 'k'den daha büyüktür. |
---|
public static ApproxTopK <T> create ( Kapsam kapsamı, İşlenen <T> girişi, Uzun k, Seçenekler... seçenekler)
Yeni bir ApproxTopK işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|---|
giriş | Aranacak dizi. Kayan tipin en az 1-D'si olmalıdır |
k | Min/maks-k sayısını belirtir. |
seçenekler | isteğe bağlı nitelik değerlerini taşır |
İadeler
- ApproxTopK'un yeni bir örneği
genel Çıkış <Tamsayı> endeksleri ()
"Giriş" işleneninin "reduction_dimension" boyunca "değerler" endeksleri.
genel statik ApproxTopK.Options isMaxK (Boolean isMaxK)
Parametreler
isMaxK | Doğru olduğunda max-k'yi hesaplar; aksi takdirde min-k'yi hesaplar. |
---|
public static ApproxTopK.Options geri çağırmaTarget (Float geri çağırmaTarget)
Parametreler
geri çağırmaHedefi | Yaklaşım için hedefi hatırlayın. Aralık (0,1] |
---|
genel statik ApproxTopK.Seçenek azaltmaBoyut (Uzun azaltmaBoyut)
Parametreler
küçültmeBoyut | Aranacak tamsayı boyutu. Varsayılan: -1. |
---|
public static ApproxTopK.Options azaltımıInputSizeOverride (Uzun azaltmaInputSizeOverride)
Parametreler
azaltmaInputSizeOverride | Pozitif bir değere ayarlandığında, geri çağırmanın değerlendirilmesi için "input[reduction_dim]" tarafından belirlenen boyutu geçersiz kılar. Bu seçenek, verilen 'girdi'nin SPMD'deki veya dağıtılmış işlem hatlarındaki genel hesaplamanın yalnızca bir alt kümesi olduğu ve gerçek girdi boyutunun 'giriş' şekliyle ertelenemediği durumlarda kullanışlıdır. |
---|
genel Çıkış <T> değerleri ()
"Giriş" işleneninin "redüksiyon_boyutu" boyunca min/maks k değerleri. Boyut, "reduction_dimension" dışında "giriş" işleneniyle aynıdır: "aggregate_to_topk" doğru olduğunda, indirgeme boyutu "k" olur; aksi takdirde, boyutun uygulama tarafından tanımlandığı durumda daha büyük 'k'ye eşittir.