OutfeedDequeueTupleV2

classe final pública OutfeedDequeueTupleV2

Recupere vários valores do outfeed de computação. O ordinal do dispositivo é um tensor que permite o outfeed dinâmico.

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

Métodos Públicos

static OutfeedDequeueTupleV2
criar ( Escopo do escopo, Operando <Integer> deviceOrdinal, List <Class <? >> dtypes, List < Shape > shapes)
Método de fábrica para criar uma classe envolvendo uma nova operação OutfeedDequeueTupleV2.
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 OutfeedDequeueTupleV2 create ( Escopo do escopo, Operand <Integer> deviceOrdinal, List <Class <? >> dtypes, List < Shape > shapes)

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

Parâmetros
alcance escopo atual
deviceOrdinal Um tensor escalar int, representando 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.
dtypes Os tipos de elemento de cada elemento em `saídas`.
formas As formas de cada tensor em `saídas`.
Devoluções
  • uma nova instância de OutfeedDequeueTupleV2

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

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

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