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> | |
Çıkış <T> | veri () |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
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