Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

NcclAllReduce

kelas akhir public NcclAllReduce

Mengeluarkan tensor yang berisi pengurangan di semua tensor input.

Menghasilkan tensor yang berisi pengurangan di semua tensor input yang diteruskan ke ops dalam `shared_name yang sama.

Grafik harus dibuat sehingga jika satu op berjalan dengan nilai shared_name `c`, maka `num_devices` ops akan berjalan dengan nilai shared_name `c`. Kegagalan untuk melakukannya akan menyebabkan eksekusi grafik gagal untuk diselesaikan.

input: input ke data reduksi: nilai reduksi di semua `jumlah_perangkat`. reduksi: operasi reduksi yang akan dilakukan. num_devices: Jumlah perangkat yang berpartisipasi dalam pengurangan ini. shared_name: Pengidentifikasi yang dibagikan di antara operasi pengurangan yang sama.

Metode Publik

Output <T>
asOutput ()
Mengembalikan pegangan simbolis tensor.
statis <T meluas Nomor> NcclAllReduce <T>
membuat ( Lingkup lingkup, Operan <T> input, pengurangan String, Long numDevices, String sharedName)
Metode pabrik untuk membuat kelas yang membungkus operasi NcclAllReduce baru.
Output <T>
Data ()

Metode yang Diwarisi

Metode Publik

publik Keluaran <T> asOutput ()

Mengembalikan pegangan simbolis tensor.

Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

public static NcclAllReduce <T> membuat ( Lingkup lingkup, Operan <T> input, pengurangan String, Long numDevices, String sharedName)

Metode pabrik untuk membuat kelas yang membungkus operasi NcclAllReduce baru.

Parameter
cakupan lingkup saat ini
Kembali
  • contoh baru dari NcclAllReduce

publik Keluaran <T> data ()