OutfeedDequeueTuple

classe final pública OutfeedDequeueTuple

Recupere vários valores do outfeed de computação.

Esta operação será bloqueada indefinidamente até que os dados estejam disponíveis. A saída `i` corresponde ao elemento da tupla XLA` i`.

Classes aninhadas

aula OutfeedDequeueTuple.Options Atributos opcionais para OutfeedDequeueTuple

Métodos Públicos

static OutfeedDequeueTuple
criar ( escopo de escopo, List <Class <? >> dtypes, List < Shape > shapes, Options ... options)
Método de fábrica para criar uma classe envolvendo uma nova operação OutfeedDequeueTuple.
static OutfeedDequeueTuple.Options
deviceOrdinal (Long deviceOrdinal)
Iterador < Operando <Objeto>>
Listar < Resultado <? >>
saídas ()
Uma lista de tensores que serão lidos no outfeed.

Métodos herdados

Métodos Públicos

public static OutfeedDequeueTuple create ( Escopo do escopo, List <Class <? >> dtypes, List < Shape > shapes, Options ... options)

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

Parâmetros
alcance escopo atual
dtypes Os tipos de elemento de cada elemento em `saídas`.
formas As formas de cada tensor em `saídas`.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de OutfeedDequeueTuple

public static OutfeedDequeueTuple.Options deviceOrdinal (Long deviceOrdinal)

Parâmetros
deviceOrdinal O dispositivo TPU a ser usado. Deve ser -1 quando o Op está em execução em um dispositivo TPU e> = 0 quando o Op está em execução no dispositivo de CPU.

Iterador público < Operando <Objeto>> iterador ()

Lista pública < Saída <? >> saídas ()

Uma lista de tensores que serão lidos no outfeed.