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

Genel Yöntemler

Çıkış <T>
Çıkış olarak ()
Bir tensörün sembolik tutamacını döndürür.
statik <T Sayıyı 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

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 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 ()