NcclReduce

kelas akhir publik NcclReduce

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

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

Grafik harus dibuat sedemikian rupa sehingga semua input memiliki penetapan perangkat yang valid, dan operasi itu sendiri ditetapkan ke salah satu perangkat tersebut.

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

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 > NcclReduce <T>
buat ( Lingkup lingkup, input Iterable< Operand <T>>, Pengurangan string)
Metode pabrik untuk membuat kelas yang membungkus operasi NcclReduce baru.
Keluaran <T>
data ()

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "NcclReduce"

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 NcclReduce <T> buat ( Lingkup lingkup, input Iterable< Operand <T>>, Pengurangan string)

Metode pabrik untuk membuat kelas yang membungkus operasi NcclReduce baru.

Parameter
cakupan ruang lingkup saat ini
Kembali
  • contoh baru dari NcclReduce

publik Keluaran <T> data ()