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> | |
Yineleyici< İşlenen <T>> | yineleyici () |
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