Ç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]'.
Sabitler
Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
Genel Yöntemler
Çıkış <T> | Çıkış olarak () Tensörün sembolik tutamacını döndürür. |
static <T TNumber'ı genişletir > CrossReplicaSum <T> | |
Çıkış <T> | çıktı () Dağıtılmış tüm girdilerin toplamı. |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Genel Yöntemler
genel Çıkış <T> asOutput ()
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 < TInt32 > 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