Ç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