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>> | iterador () |
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. |
---|