TPUPartitionedOutputV2

genel son sınıf TPUPartitionedOutputV2

XLA tarafından parçalanacak bir tensörün bölümlenmiş bir listeye çoğaltılmasını sağlayan bir işlem

XLA hesaplamasının dışındaki çıktılar. ND parçalamayı destekler.

Genel Yöntemler

statik <T> TPUPartitionedOutputV2 <T>
create ( Kapsam kapsamı, İşlenen <T> girişleri, Uzun numSplits, List<Long> partitionDims)
Yeni bir TPUPartitionedOutputV2 işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Yineleyici< İşlenen <T>>
Liste< Çıkış <T>>
çıktı ()
Aynı şekle sahip bölümlenmiş çıktıların listesi.

Kalıtsal Yöntemler

Genel Yöntemler

public static TPUPartitionedOutputV2 <T> create ( Kapsam kapsamı, İşlenen <T> girişleri, Uzun numSplits, List<Long> partitionDims)

Yeni bir TPUPartitionedOutputV2 işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Parametreler
kapsam mevcut kapsam
girişler Bölümlenmiş tensörlerin tam şeklini temsil eden bir tensör.
bölümDims Her boyutun nasıl bölümlendiğini açıklayan tam sayıların listesi. Boşluk, girdilerin kopyalandığını gösterir.
İadeler
  • TPUPartitionedOutputV2'nin yeni bir örneği

public Yineleyici< İşlenen <T>> yineleyici ()

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

Aynı şekle sahip bölümlenmiş çıktıların listesi.