CollectivePermute

genel final dersi CollectivePermute

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

Sabit Değer: "Toplu Permute"

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

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

Permütasyonlu giriş.