NcclAllReduce

kelas akhir publik NcclAllReduce

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

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

publik Keluaran <T> data ()