TPUPartitionedOutput

public class akhir TPUPartitionedOutput

Sebuah op yang mendemultipleks tensor untuk di-sharding oleh XLA ke daftar yang dipartisi

output di luar perhitungan XLA.

Kelas Bersarang

kelas

TPUPartitionedOutput.Options

Atribut opsional untuk TPUPartitionedOutput

Metode Publik

statis <T> TPUPartitionedOutput <T>	membuat ( Lingkup lingkup, Operan <T> masukan, Long numSplits, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutput baru.
Iterator < Operan <T >>	iterator ()
Daftar < Keluaran <T >>	Output () Daftar input yang dipartisi yang harus memiliki bentuk yang sama.
statis TPUPartitionedOutput.Options	partitionDim (Long partitionDim)

Metode yang Diwarisi

Dari kelas org.tensorflow.op.PrimitiveOp

boolean akhir	sama (Object obj)
akhir	hashCode ()
Operasi	op () Mengembalikan mendasari Operation
string terakhir	toString ()

Dari kelas java.lang.Object

boolean	sama (Object arg0)
Kelas terakhir<?>	getClass ()
ke dalam	hashCode ()
kekosongan akhir	notify ()
kekosongan akhir	notifyAll ()
Rangkaian	toString ()
kekosongan akhir	menunggu (arg0 panjang, int arg1)
kekosongan akhir	menunggu (arg0 panjang)
kekosongan akhir	menunggu ()

Dari antarmuka java.lang.Iterable

kekosongan abstrak	forEach (Consumer <? Super T> arg0)
abstrak Iterator < Operan <T >>	iterator ()
abstrak Spliterator < Operan <T >>	spliterator ()

Metode Publik

public static TPUPartitionedOutput <T> membuat ( Lingkup lingkup, Operan <T> masukan, Long numSplits, Options ... pilihan)

Metode pabrik untuk membuat kelas yang membungkus operasi TPUPartitionedOutput baru.

Parameter

cakupan	lingkup saat ini
masukan	Tensor yang mewakili bentuk penuh dari tensor yang dipartisi.
pilihan	membawa nilai atribut opsional

Kembali

• contoh baru dari TPUPartitionedOutput

public Iterator < Operan <T >> iterator ()

public List < Keluaran <T >> keluaran ()

Daftar input yang dipartisi yang harus memiliki bentuk yang sama.

public static TPUPartitionedOutput.Options partitionDim (Long partitionDim)

Parameter

partisiDim	Sebuah bilangan bulat menggambarkan dimensi mana yang dipartisi.