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.
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
| Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
| statis <T memperluas TNomber > NcclAllReduce <T> | buat ( Lingkup lingkup , masukan Operan <T>, Pengurangan string, Jumlah Perangkat yang panjang, String sharedName) Metode pabrik untuk membuat kelas yang membungkus operasi NcclAllReduce baru. |
| Keluaran <T> | data () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolis 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