Recupera los 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 opaco que se puede guardar y restaurar de una manera compatible con ejecuciones futuras.
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. 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. 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. 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 caída. 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 | CudnnRNNParamsToCanonicalV2.Options | Atributos opcionales para CudnnRNNParamsToCanonicalV2 |
Métodos públicos
Lista < Salida <T>> | sesgos () |
estática <T extiende Número> CudnnRNNParamsToCanonicalV2 <T> | crear ( Alcance alcance, operando <entero> numLayers, operando <Integer> numUnits, operando <Integer> inputSize, operando <T> params, numParamsWeights mucho, mucho numParamsBiases, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación CudnnRNNParamsToCanonicalV2. |
static CudnnRNNParamsToCanonicalV2.Options | dirección ( dirección de la cuerda) |
static CudnnRNNParamsToCanonicalV2.Options | |
static CudnnRNNParamsToCanonicalV2.Options | inputMode (String inputMode) |
static CudnnRNNParamsToCanonicalV2.Options | |
static CudnnRNNParamsToCanonicalV2.Options | rnnMode (Cadena rnnMode) |
static CudnnRNNParamsToCanonicalV2.Options | |
static CudnnRNNParamsToCanonicalV2.Options | seed2 (Semilla larga2) |
Lista < Salida <T>> | pesos () |
Métodos heredados
Métodos públicos
public static CudnnRNNParamsToCanonicalV2 <T> Crear ( Alcance alcance, operando <entero> numLayers, operando <entero> numUnits, operando <Integer> inputSize, operando <T> params, numParamsWeights mucho, mucho numParamsBiases, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación CudnnRNNParamsToCanonicalV2.
Parámetros
alcance | alcance actual |
---|---|
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de CudnnRNNParamsToCanonicalV2