CudnnRNNCanonicalToParamsV2

CudnnRNNCanonicalToParamsV2 public final class

Convierte los parámetros de CudnnRNN de forma canónica a forma utilizable. Soporta la proyección en LSTM.

Escribe un conjunto de pesos en el búfer de parámetros opacos para que se puedan usar en próximos entrenamientos o inferencias.

Tenga en cuenta que es posible que el búfer de parámetros no sea compatible con diferentes GPU. Por lo tanto, cualquier guardado y restauración debe convertirse hacia y desde los pesos y sesgos canónicos.

num_layers: especifica el número de capas en el modelo RNN. num_units: especifica el tamaño del estado oculto. input_size: especifica el tamaño del estado de entrada. pesos: la forma canónica de pesos que se puede utilizar para guardar y restaurar. Es más probable que sean compatibles entre diferentes generaciones. sesgos: la forma canónica de sesgos que se puede utilizar para salvar y restaurar. Es más probable que sean compatibles entre diferentes generaciones. num_params_weights: número de matriz de parámetros de peso para todas las capas. num_params_biases: número de vector de parámetro de sesgo para todas las capas. rnn_mode: Indica el tipo de modelo RNN. input_mode: indica si hay una proyección lineal entre la entrada y el cálculo real antes de la primera capa. 'skip_input' solo se permite cuando input_size == num_units; 'auto_select' implica 'skip_input' cuando input_size == num_units; de lo contrario, implica 'linear_input'. direction: Indica si se utilizará un modelo bidireccional. dir = (dirección == bidireccional)? Abandono 2: 1: probabilidad de abandono. Cuando se establece en 0., la deserción está deshabilitada. semilla: la primera parte de una semilla para inicializar la deserción. seed2: la segunda parte de una semilla para inicializar la deserción. num_proj: la dimensionalidad de salida para las matrices de proyección. Si Ninguno o 0, no se realiza ninguna proyección.

Clases anidadas

clase CudnnRNNCanonicalToParamsV2.Options Los atributos opcionales para CudnnRNNCanonicalToParamsV2

Métodos públicos

Salida <T>
asOutput ()
Devuelve el identificador simbólico de un tensor.
estática <T se extiende Número> CudnnRNNCanonicalToParamsV2 <T>
crear ( Alcance alcance, operando <entero> numLayers, operando <entero> numUnits, operando <Integer> inputSize, Iterable < operando <T >> pesos, Iterable < operando <T >> sesgos, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación CudnnRNNCanonicalToParamsV2.
estáticas CudnnRNNCanonicalToParamsV2.Options
dirección (dirección String)
estáticas CudnnRNNCanonicalToParamsV2.Options
deserción (deserción Float)
estáticas CudnnRNNCanonicalToParamsV2.Options
InputMode (String InputMode)
estáticas CudnnRNNCanonicalToParamsV2.Options
numProj (Long numProj)
Salida <T>
params ()
estáticas CudnnRNNCanonicalToParamsV2.Options
rnnMode (String rnnMode)
estáticas CudnnRNNCanonicalToParamsV2.Options
semilla (semilla de larga duración)
estáticas CudnnRNNCanonicalToParamsV2.Options
seed2 (Long seed2)

Métodos heredados

Métodos públicos

pública de salida <T> asOutput ()

Devuelve el identificador simbólico de un tensor.

Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.

public static CudnnRNNCanonicalToParamsV2 <T> crear ( Alcance alcance, operando <Integer> numLayers, operando <Integer> numUnits, operando <Integer> inputSize, Iterable < operando <T >> pesos, Iterable < operando <T >> sesgos, Opciones .. . opciones)

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

Parámetros
alcance alcance actual
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de CudnnRNNCanonicalToParamsV2

public static CudnnRNNCanonicalToParamsV2.Options dirección (dirección de cuerdas)

estáticas pública CudnnRNNCanonicalToParamsV2.Options de abandono (deserción Float)

public static CudnnRNNCanonicalToParamsV2.Options InputMode (String InputMode)

public static CudnnRNNCanonicalToParamsV2.Options numProj (Long numProj)

pública de salida <T> params ()

public static CudnnRNNCanonicalToParamsV2.Options rnnMode (String rnnMode)

public static CudnnRNNCanonicalToParamsV2.Options semilla (semilla de larga duración)

public static CudnnRNNCanonicalToParamsV2.Options seed2 (Long seed2)