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