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
estática OutfeedDequeueTupleV2 | |
Iterator < Operando <Object >> | iteração () |
List < Output <? >> | saídas () Uma lista de tensores que serão lidos no outfeed. |
Métodos herdados
Métodos Públicos
public static OutfeedDequeueTupleV2 criar ( Scope escopo, Operando <Integer> deviceOrdinal, List <Class <? >> dtypes, List < Forma > formas)
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