CrossReplicaSum

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]'.

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>
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.
Çı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ı

Sabit Değer: "CrossReplicaSum"

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

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

Dağıtılmış tüm girdilerin toplamı.