Menghasilkan tensor yang berisi reduksi di semua tensor masukan.
Menghasilkan tensor yang berisi pengurangan seluruh tensor masukan yang diteruskan ke operasi dalam `nama_bersama.
Grafik harus dibuat sehingga jika satu operasi berjalan dengan nilai nama_bersama `c`, maka operasi `num_devices` akan berjalan dengan nilai nama_bersama `c`. Kegagalan untuk melakukannya akan menyebabkan eksekusi grafik gagal diselesaikan.
input: masukan ke data pengurangan: nilai pengurangan di semua `num_devices` perangkat. reduksi: operasi reduksi yang akan dilakukan. num_devices: Jumlah perangkat yang berpartisipasi dalam pengurangan ini. shared_name: Pengidentifikasi yang dibagikan antar operasi pengurangan yang sama.
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
statis <T memperluas Nomor> NcclAllReduce <T> | |
Keluaran <T> | data () |
Metode Warisan
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static NcclAllReduce <T> buat ( Lingkup lingkup, masukan Operan <T>, Pengurangan string, Jumlah Perangkat Panjang, String sharedName)
Metode pabrik untuk membuat kelas yang membungkus operasi NcclAllReduce baru.
Parameter
cakupan | ruang lingkup saat ini |
---|
Kembali
- contoh baru dari NcclAllReduce