TPUPartitionedOutputV2

classe final pública TPUPartitionedOutputV2

Uma operação que demultiplexa um tensor para ser fragmentado por XLA em uma lista de

saídas fora da computação XLA. Suporta fragmentação ND.

Métodos públicos

static <T> TPUPartitionedOutputV2 <T>	create ( Escopo do escopo , entradas do Operando <T>, Long numSplits, List<Long> partitionDims) Método de fábrica para criar uma classe envolvendo uma nova operação TPUPartitionedOutputV2.
Iterador < Operando <T>>	iterador ()
Lista< Saída <T>>	saída () Uma lista de saídas particionadas que têm a mesma forma.

Métodos Herdados

Da classe org.tensorflow.op.PrimitiveOp

final booleano	é igual a (Objeto obj)
int final	hashCode ()
Operação	op () Retorna a Operation subjacente
String final	toString ()

Da classe java.lang.Object

boleano	igual a (objeto arg0)
Aula final<?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notificarTodos ()
Corda	toString ()
vazio final	espera (long arg0, int arg1)
vazio final	espera (arg0 longo)
vazio final	espere ()

Da interface java.lang.Iterable

vazio	forEach (Consumidor<? super T> arg0)
abstract Iterador < Operando <T>>	iterador ()
Divisor < Operando <T>>	divisor ()

Métodos públicos

public static TPUPartitionedOutputV2 <T> create ( Scope scope, Operand <T> inputs, Long numSplits, List<Long> partitionDims)

Método de fábrica para criar uma classe envolvendo uma nova operação TPUPartitionedOutputV2.

Parâmetros

escopo	escopo atual
entradas	Um tensor que representa a forma completa dos tensores particionados.
partiçãoDims	Uma lista de inteiros descrevendo como cada dimensão é particionada. O vazio indica que as entradas são replicadas.

devoluções

• uma nova instância de TPUPartitionedOutputV2

public Iterator< Operand <T>> iterator ()

public List< Output <T>> output ()

Uma lista de saídas particionadas que têm a mesma forma.