InfeedEnqueue

InfeedEnqueue public final class

Una operación que introduce un único valor de Tensor en el cálculo.

Clases anidadas

clase InfeedEnqueue.Options Los atributos opcionales para InfeedEnqueue

Métodos públicos

estática <T> InfeedEnqueue
crear ( Alcance alcance, operando <T> de entrada, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación InfeedEnqueue.
estáticas InfeedEnqueue.Options
deviceOrdinal (Long deviceOrdinal)
estáticas InfeedEnqueue.Options
diseño (List <Larga> diseño)
estáticas InfeedEnqueue.Options
forma ( Shape forma)

Métodos heredados

Métodos públicos

public static InfeedEnqueue crear ( Alcance alcance, operando <T> de entrada, Opciones ... Opciones)

Método de fábrica para crear una clase que envuelva una nueva operación InfeedEnqueue.

Parámetros
alcance alcance actual
aporte Un tensor que se proporcionará mediante el mecanismo de alimentación.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de InfeedEnqueue

public static InfeedEnqueue.Options deviceOrdinal (Long deviceOrdinal)

Parámetros
deviceOrdinal El dispositivo de TPU que se utilizará. Debe ser -1 cuando el Op se está ejecutando en un dispositivo de TPU y> = 0 cuando el Op se está ejecutando en el dispositivo de la CPU.

public static InfeedEnqueue.Options diseño (List <Larga> diseño)

Parámetros
diseño Un vector que contiene el diseño solicitado en una secuencia de menor a mayor. Si se pasa un atributo de diseño, pero sus valores son todos -1, la operación de alimentación calculará el diseño.

estáticas pública InfeedEnqueue.Options forma ( Shape forma)

Parámetros
forma La forma del tensor.