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

NcclReduce

kelas akhir public NcclReduce

Mengurangi `input` dari `num_devices` menggunakan `reduction` ke satu perangkat.

Mengurangi `input` dari `num_devices` menggunakan `reduction` ke satu perangkat.

Grafik harus dibangun sehingga semua input memiliki penetapan perangkat yang valid, dan op itu sendiri diberikan salah satu perangkat ini.

masukan: Masukan untuk pengurangan. data: nilai pengurangan di semua perangkat `num_devices`. reduksi: operasi reduksi yang akan dilakukan.

Metode Publik

Output <T>
asOutput ()
Mengembalikan pegangan simbolis tensor.
statis <T meluas Nomor> NcclReduce <T>
membuat ( Lingkup lingkup, Iterable < Operan <T >> masukan, pengurangan String)
Metode pabrik untuk membuat kelas yang membungkus operasi NcclReduce 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 NcclReduce <T> membuat ( Lingkup lingkup, Iterable < Operan <T >> masukan, pengurangan String)

Metode pabrik untuk membuat kelas yang membungkus operasi NcclReduce baru.

Parameter
cakupan lingkup saat ini
Kembali
  • contoh baru dari NcclReduce

publik Keluaran <T> data ()