TPUPartitionedOutputV2

TPUPartitionedOutputV2 kelas akhir publik

Sebuah operasi yang mendemultiplexing tensor untuk dipecah oleh XLA ke daftar yang dipartisi

keluaran di luar komputasi XLA. Mendukung sharding ND.

Metode Publik

statis <T> TPUPartitionedOutputV2 <T>	buat ( Lingkup lingkup, input Operand <T>, Long numSplits, Daftar<Panjang> partisiDims) Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutputV2 baru.
Iterator< Operan <T>>	pengulangan ()
Daftar< Keluaran <T>>	keluaran () Daftar keluaran yang dipartisi yang memiliki bentuk yang sama.

Metode Warisan

Dari kelas org.tensorflow.op.PrimitiveOp

boolean terakhir	sama dengan (Objek objek)
int terakhir	Kode hash ()
Operasi	op () Mengembalikan Operation yang mendasarinya
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 java.lang.Iterable

ruang kosong	forEach (Konsumen<? super T> arg0)
abstrak Iterator< Operan <T>>	pengulangan ()
Pemisah< Operan <T>>	pemisah ()

Metode Publik

public static TPUPartitionedOutputV2 <T> buat ( Lingkup lingkup, input Operan <T>, numSplits panjang, Daftar<Panjang> partisiDims)

Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutputV2 baru.

Parameter

cakupan	ruang lingkup saat ini
masukan	Tensor yang mewakili bentuk penuh tensor yang dipartisi.
partisiDim	Daftar bilangan bulat yang menjelaskan bagaimana setiap dimensi dipartisi. Kekosongan menunjukkan masukan direplikasi.

Kembali

• contoh baru TPUPartitionedOutputV2

Iterator publik< Operan <T>> iterator ()

Daftar publik< Keluaran <T>> keluaran ()

Daftar keluaran yang dipartisi yang memiliki bentuk yang sama.