halka açık final sınıfı CrossReplicaSum
Çoğaltılmış TPU örnekleri genelinde girişleri toplamak için bir Op.
Her örnek kendi girdisini sağlar.
Örneğin 8 TPU örneğinin olduğunu varsayalım: "[A, B, C, D, E, F, G, H]". group_questment=`[[0,2,4,6],[1,3,5,7]]' iletilmesi, 'A, C, E, G'yi grup 0 ve 'B, D, F, H' olarak ayarlar Grup 1 olarak şu çıktıları alıyoruz: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]'.
Genel Yöntemler
| Çıkış <T> | Çıkış olarak () Bir tensörün sembolik tutamacını döndürür. |
| statik <T Sayıyı genişletir> CrossReplicaSum <T> | |
| Çıkış <T> | çıktı () Dağıtılmış tüm girdilerin toplamı. |
Kalıtsal Yöntemler
Genel Yöntemler
genel Çıkış <T> asOutput ()
Bir tensörün sembolik tutamacını döndürür.
TensorFlow işlemlerinin girdileri, başka bir TensorFlow işleminin çıktılarıdır. Bu yöntem, girişin hesaplanmasını temsil eden sembolik bir tanıtıcı elde etmek için kullanılır.
public static CrossReplicaSum <T> create ( Kapsam kapsamı , İşlenen <T> girişi, İşlenen <Tamsayı> grupAtaması)
Yeni bir CrossReplicaSum işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
| kapsam | mevcut kapsam |
|---|---|
| giriş | Toplamın yerel girişi. |
| grup ödevi | [num_groups, num_replicas_per_group] şeklinde bir int32 tensörü. 'group_questment[i]', i'inci alt gruptaki kopya kimliklerini temsil eder. |
İadeler
- CrossReplicaSum'un yeni bir örneği