Uyarı: Bu API kullanımdan kaldırıldı ve TensorFlow'un gelecekteki bir sürümünde , değişim kararlı hale geldikten sonra kaldırılacaktır.

Bu sayfa, Cloud Translation API ile çevrilmiştir.
Switch to English

AllToAll

genel final sınıfı AllToAll

TPU kopyaları arasında veri alışverişi yapmak için bir İşlem.

Her bir eşlemede, girdi "bölme_boyutu" boyunca "bölme_sayısı" bloklarına bölünür ve grup ataması verilen diğer eşlemelere gönderilir. Diğer kopyalardan "split_count" - 1 blok aldıktan sonra, çıktı olarak "concat_dimension" boyunca blokları birleştiririz.

Örneğin, 2 TPU kopyası olduğunu varsayalım: kopya 0 giriş alır: "[[A, B]]" kopya 1 şu girişi alır: "[[C, D]]"

group_assignment = "[[0, 1]]` concat_dimension = 0 split_dimension = 1 split_count = 2

kopya 0'ın çıkışı: "[[A], [C]]" kopya 1'in çıkışı: "[[B], [D]]`

Kamusal Yöntemler

Çıkış <T>
asOutput ()
Bir tensörün sembolik tutamacını döndürür.
statik <T> AllToAll <T>
oluşturmak ( Kapsam kapsamı, Operand <T> girişi, Operand <tamsayı> groupAssignment, Uzun concatDimension, Uzun splitDimension, uzun splitCount)
Yeni bir AllToAll işlemini saran bir sınıf oluşturmak için fabrika yöntemi.
Çıkış <T>
çıktı ()
Değiştirilen sonuç.

Devralınan Yöntemler

Kamusal Yöntemler

public Output <T> asOutput ()

Bir tensörün sembolik tutamacını döndürür.

TensorFlow işlemlerine yönelik girdiler, başka bir TensorFlow işleminin çıktılarıdır. Bu yöntem, girdinin hesaplanmasını temsil eden sembolik bir tutamaç elde etmek için kullanılır.

public static AllToAll <T> create ( Scope kapsamı, Operand <T> input, Operand <Integer> groupAssignment, Long concatDimension, Long splitDimension, Long splitCount)

Yeni bir AllToAll işlemini saran bir sınıf oluşturmak için fabrika yöntemi.

Parametreler
dürbün mevcut kapsam
giriş Toplamın yerel girdisi.
grup ödevi [Num_groups, num_replicas_per_group] şeklinde bir int32 tensörü. "grup_ ataması [i]" i. alt gruptaki kopya kimliklerini temsil eder.
concatDimension Birleştirilecek boyut numarası.
splitDimension Bölünecek boyut numarası.
splitCount Bölme sayısı, bu sayı alt grup boyutuna eşit olmalıdır (group_assignment.get_shape () [1])
İadeler
  • AllToAll'ın yeni bir örneği

public Output <T> çıkışı ()

Değiştirilen sonuç.