OutfeedDequeueTuple

OutfeedDequeueTuple classe final pública

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

classe OutfeedDequeueTuple.Options Atributos opcionais para OutfeedDequeueTuple

Métodos Públicos

estática OutfeedDequeueTuple
criar ( Scope escopo, List <Class <? >> dtypes, List < Forma > formas, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação OutfeedDequeueTuple.
estáticos OutfeedDequeueTuple.Options
deviceOrdinal (Long deviceOrdinal)
Iterator < Operando <Object >>
List < Output <? >>
saídas ()
Uma lista de tensores que serão lidos no outfeed.

Métodos herdados

Métodos Públicos

public static OutfeedDequeueTuple criar ( Scope escopo, List <Class <? >> dtypes, List < Forma > formas, Options ... Opções)

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.

pública Iterator < Operando <Object >> iterator ()

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

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