Recupera parámetros de CudnnRNN en forma canónica. Soporta la proyección en LSTM.
Recupera un conjunto de pesos del búfer de parámetros opacos que se pueden guardar y restaurar de forma compatible con ejecuciones futuras.
Tenga en cuenta que el búfer de parámetros puede no ser 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. num_params_weights: número de matriz de parámetros de peso para todas las capas. num_params_biases: número de vectores de parámetros de sesgo para todas las capas. 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 el ahorro y la restauración. Es más probable que sean compatibles entre diferentes generaciones. 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 está permitido cuando input_size == num_units; 'auto_select' implica 'skip_input' cuando input_size == num_units; de lo contrario, implica 'entrada_lineal'. direction: Indica si se utilizará un modelo bidireccional. dir = (dirección == bidireccional) ? 2 : 1 abandono: probabilidad de abandono. Cuando se establece en 0, el abandono está deshabilitado. semilla: la primera parte de una semilla para inicializar el abandono. seed2: la segunda parte de una semilla para inicializar el abandono. num_proj: La dimensionalidad de salida para las matrices de proyección. Si es Ninguno o 0, no se realiza ninguna proyección.
Clases anidadas
clase | CudnnRNNParamsToCanonicalV2.Opciones | Atributos opcionales para CudnnRNNParamsToCanonicalV2 |
Métodos públicos
Lista< Salida <T>> | sesgos () |
estático <T extiende Número> CudnnRNNParamsToCanonicalV2 <T> | create ( Alcance alcance, Operando <Entero> numCapas, Operando <Entero> numUnidades, Operando <Entero> inputSize, Operando <T> params, Long numParamsWeights, Long numParamsBiases, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación CudnnRNNParamsToCanonicalV2. |
CudnnRNNParamsToCanonicalV2.Options estático | dirección (dirección de la cadena) |
CudnnRNNParamsToCanonicalV2.Options estático | abandono (abandono flotante) |
CudnnRNNParamsToCanonicalV2.Options estático | modo de entrada (Cadena modo de entrada) |
CudnnRNNParamsToCanonicalV2.Options estático | numProj (NúmProy largo) |
CudnnRNNParamsToCanonicalV2.Options estático | rnnMode (Cadena rnnMode) |
CudnnRNNParamsToCanonicalV2.Options estático | semilla (semilla larga) |
CudnnRNNParamsToCanonicalV2.Options estático | semilla2 (semilla larga2) |
Lista< Salida <T>> | pesos () |
Métodos Heredados
Métodos públicos
public static CudnnRNNParamsToCanonicalV2 <T> create ( Scope scope, Operand <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Operand <T> params, Long numParamsWeights, Long numParamsBiases, Options... options)
Método de fábrica para crear una clase que envuelve una nueva operación CudnnRNNParamsToCanonicalV2.
Parámetros
alcance | alcance actual |
---|---|
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de CudnnRNNParamsToCanonicalV2