NcclAllReduce

genel final sınıfı NcclAllReduce

Tüm giriş tensörlerindeki azalmayı içeren bir tensör çıkışı sağlar.

Aynı `paylaşılan_adı' içindeki işlemlere iletilen tüm giriş tensörlerindeki azalmayı içeren bir tensör çıktısı alır.

Grafik, eğer bir operasyon paylaşılan_adı değeri 'c' ile çalışıyorsa, 'num_devices' operasyonları paylaşılan_adı değeri 'c' ile çalışacak şekilde oluşturulmalıdır. Bunun yapılmaması, grafiğin yürütülmesinin tamamlanmamasına neden olur.

giriş: azaltma verilerinin girişi: tüm "num_devices" aygıtlarındaki azaltmanın değeri. azaltma: gerçekleştirilecek azaltma işlemi. num_devices: Bu azalmaya katılan cihazların sayısı. paylaşılan_adı: Aynı azaltmanın operasyonları arasında paylaşılan tanımlayı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, TNumber'ı genişletir > NcclAllReduce <T>
create ( Kapsam kapsamı, İşlenen <T> girişi, Dize azaltma, Uzun numDevices, String paylaşılanAd)
Yeni bir NcclAllReduce işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Çıkış <T>
veri ()

Kalıtsal Yöntemler

Sabitler

genel statik son Dize OP_NAME

Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı

Sabit Değer: "NcclAllReduce"

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 NcclAllReduce <T> create ( Kapsam kapsamı, İşlenen <T> girişi, Dize azaltma, Uzun numDevices, String paylaşılanAd)

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

Parametreler
kapsam mevcut kapsam
İadeler
  • yeni bir NcclAllReduce örneği

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