Çoğaltılmış TPU örneklerinde tensörlerin izinini vermeye yönelik bir Op.
Her örnek kendi girdisini sağlar.
Örneğin 4 TPU örneğinin olduğunu varsayalım: "[A, B, C, D]". source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]' iletilmesi şu çıktıları alır: `[D, A, B, C]'.
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, TType'ı genişletir > CollectivePermute <T> | |
Çıkış <T> | çıktı () Permütasyonlu giriş. |
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 CollectivePermute <T> create ( Kapsam kapsamı, İşlenen <T> girişi, İşlenen < TInt32 > sourceTargetPairs)
Yeni bir CollectivePermute işlemini saran bir sınıf oluşturmak için fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|---|
giriş | İzin verilecek yerel giriş. Şu anda yalnızca float ve bfloat16'yı desteklemektedir. |
kaynakHedefÇiftleri | [num_pairs, 2] şeklinde bir tensör. |
İadeler
- CollectivePermute'un yeni bir örneği