Aprenda o que há de mais recente em aprendizado de máquina, IA generativa e muito mais no WiML Symposium 2023 Registre-se

Esta página foi traduzida pela API Cloud Translation.

TPUPartitionedOutput

classe final pública TPUPartitionedOutput

Um op que demultiplexa um tensor a ser fragmentado pelo XLA para uma lista de partições

saídas fora da computação XLA.

aula TPUPartitionedOutput.Options Atributos opcionais para TPUPartitionedOutput

estático <T> TPUPartitionedOutput <T>	create ( Escopo do escopo, entradas Operando <T>, Long numSplits, Opções... opções) Método de fábrica para criar uma classe que envolve uma nova operação TPUPartitionedOutput.
Iterador< Operando <T>>	iterador ()
Lista< Saída <T>>	saída () Uma lista de entradas particionadas que devem ter a mesma forma.
estático TPUPartitionedOutput.Options	partitionDim (Partição longa Dim)

booleano final	igual a (objeto obj)
final int	código de hash ()
Operação	op () Retorna a `Operation` subjacente
String final	toString ()

Da classe java.lang.Object

Da interface java.lang.Iterable

vazio	forEach (Consumidor<? super T> arg0)
abstrato Iterador< Operando <T>>	iterador ()
Divisor < Operando <T>>	separador ()

Métodos públicos

Método de fábrica para criar uma classe que envolve uma nova operação TPUPartitionedOutput.

alcance	escopo atual
entradas	Um tensor que representa a forma completa de tensores particionados.
opções	carrega valores de atributos opcionais

Uma lista de entradas particionadas que devem ter a mesma forma.

partitionDim	Um número inteiro descreve qual dimensão é particionada.