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.

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

Dari kelas org.tensorflow.op.RawOp

boolean terakhir	sama dengan (Objek objek)
int terakhir	Kode hash ()
Operasi	op () Kembalikan unit komputasi ini sebagai satu Operation .
String terakhir	keString ()

Dari kelas java.lang.Object

boolean	sama dengan (Objek arg0)
Kelas terakhir<?>	dapatkan Kelas ()
ke dalam	Kode hash ()
kekosongan terakhir	memberitahu ()
kekosongan terakhir	beri tahuSemua ()
Rangkaian	keString ()
kekosongan terakhir	tunggu (arg0 panjang, int arg1)
kekosongan terakhir	tunggu (argumen panjang0)
kekosongan terakhir	Tunggu ()

Dari antarmuka org.tensorflow.op.Op

Lingkungan Eksekusi abstrak	env () Kembalikan lingkungan eksekusi tempat operasi ini dibuat.
Operasi abstrak	op () Kembalikan unit komputasi ini sebagai satu Operation .

Dari antarmuka org.tensorflow.Operand

Keluaran abstrak <T>	sebagai Keluaran () Mengembalikan pegangan simbolis tensor.
abstrak T	sebagai Tensor () Mengembalikan tensor pada operan ini.
Bentuk abstrak	membentuk () Mengembalikan bentuk tensor (yang mungkin diketahui sebagian) yang dirujuk oleh Output operan ini.
Kelas abstrak<T>	jenis () Mengembalikan tipe tensor operan ini

Dari antarmuka org.tensorflow.ndarray.Shaped

abstrak ke dalam	peringkat ()
Bentuk abstrak	membentuk ()
abstrak panjang	ukuran () Menghitung dan mengembalikan ukuran total wadah ini, dalam jumlah nilai.

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "NcclAllReduce"

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