AllToAll

herkese açık final dersi HepsindenHerkese

TPU kopyaları arasında veri alışverişi yapmaya yönelik bir Op.

Her kopyada giriş, "split_dimension" boyunca "split_count" bloklarına bölünür ve grup_atama verilen diğer kopyalara gönderilir. Diğer replikalardan `split_count` - 1 blok aldıktan sonra çıktı olarak blokları `concat_dimension` boyunca birleştiriyoruz.

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

group_questment=`[[0, 1]]` concat_dimension=0 split_dimension=1 split_count=2

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

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.
statik <T, TType'ı genişletir > TümüTümüne <T>	create ( Kapsam kapsamı, İşlenen <T> girişi, İşlenen < TInt32 > groupAssignment, Long concatDimension, Long splitDimension, Long 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ç.

Kalıtsal Yöntemler

org.tensorflow.op.RawOp sınıfından

son boole değeri	eşittir (Nesne nesnesi)
son int	hash kodu ()
Operasyon	işlem () Bu hesaplama birimini tek bir Operation olarak döndürün.
son Dize	toString ()

Java.lang.Object sınıfından

boolean	eşittir (Nesne arg0)
son Sınıf<?>	getClass ()
int	hash kodu ()
son boşluk	bildir ()
son boşluk	tümünü bildir ()
Sicim	toString ()
son boşluk	bekle (uzun arg0, int arg1)
son boşluk	bekle (uzun arg0)
son boşluk	Beklemek ()

org.tensorflow.op.Op arayüzünden

özet Yürütme Ortamı	env () Bu operasyonun oluşturulduğu yürütme ortamını döndürün.
soyut Operasyon	işlem () Bu hesaplama birimini tek bir Operation olarak döndürün.

org.tensorflow.Operand arayüzünden

özet Çıkış <T>	Çıkış olarak () Tensörün sembolik tutamacını döndürür.
soyut T	Tensör olarak () Bu işlenendeki tensörü döndürür.
soyut Şekil	şekil () Bu işlenenin Output tarafından başvurulan tensörün (muhtemelen kısmen bilinen) şeklini döndürür.
soyut Sınıf<T>	tip () Bu işlenenin tensör türünü döndürür

org.tensorflow.ndarray.Shaped arayüzünden

soyut int	rütbe ()
soyut Şekil	şekil ()
soyut uzun	boyut () Bu kabın toplam boyutunu değer sayısı cinsinden hesaplar ve döndürür.

Sabitler

genel statik son Dize OP_NAME

Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı

Sabit Değer: "HepsidenHerkese"

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 AllToAll <T> create ( Kapsam kapsamı, İşlenen <T> girişi, İşlenen < TInt32 > 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

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.
concatBoyut	Birleştirilecek boyut numarası.
bölünmüş Boyut	Bölünecek boyut numarası.
splitCount	Bölünme sayısı, bu sayı alt grup boyutuna eşit olmalıdır(group_questment.get_shape()[1])

İadeler

• AllToAll'ın yeni bir örneği

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

Değiştirilen sonuç.