Ç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]'.
Genel Yöntemler
Çıkış <T> | Çıkış olarak () Bir tensörün sembolik tutamacını döndürür. |
statik <T> KollektifPermute <T> | |
Çıkış <T> | çıktı () Permütasyonlu giriş. |
Kalıtsal Yöntemler
Genel Yöntemler
genel Çıkış <T> asOutput ()
Bir 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 <Tamsayı> kaynakTargetPairs)
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