Calcula o tamanho dos pesos que podem ser usados por um modelo Cudnn RNN.
Retorne o tamanho dos parâmetros que pode ser usado pelo modelo Cudnn RNN. A alocação de peso subsequente e a inicialização devem usar esse tamanho.
num_layers: especifica o número de camadas no modelo RNN. num_units: especifica o tamanho do estado oculto. input_size: especifica o tamanho do estado de entrada. rnn_mode: indica o tipo do modelo RNN. input_mode: indica se há uma projeção linear entre a entrada e o cálculo real antes da primeira camada. 'skip_input' só é permitido quando input_size == num_units; 'auto_select' implica 'skip_input' quando input_size == num_units; caso contrário, implica em 'entrada_linear'. direção: indica se um modelo bidirecional será usado. dir = (direção == bidirecional)? 2: 1 abandono: probabilidade de abandono. Quando definido como 0., o dropout é desabilitado. semente: a 1ª parte de uma semente para inicializar o dropout. seed2: a 2ª parte de um seed para inicializar o dropout. params_size: O tamanho do buffer params que deve ser alocado e inicializado para este modelo RNN. Observe que este buffer de parâmetros pode não ser compatível com GPUs. Use CudnnRNNParamsWeights e CudnnRNNParamsBiases para salvá-los e restaurá-los de uma maneira que seja compatível em diferentes execuções.
Classes aninhadas
classe | CudnnRnnParamsSize.Options | Atributos opcionais para CudnnRnnParamsSize |
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
Output <U> | asOutput () Retorna o identificador simbólico do tensor. |
estática <U estende TNumber , T estende TNumber > CudnnRnnParamsSize <U> | |
estáticos CudnnRnnParamsSize.Options | direcção (direcção da corda) |
estáticos CudnnRnnParamsSize.Options | abandono (abandono do flutuador) |
estáticos CudnnRnnParamsSize.Options | inputMode (String inputMode) |
estáticos CudnnRnnParamsSize.Options | numProj (Long numProj) |
Output <U> | paramsSize () |
estáticos CudnnRnnParamsSize.Options | rnnMode (String rnnMode) |
estáticos CudnnRnnParamsSize.Options | sementes (semente Long) |
estáticos CudnnRnnParamsSize.Options | seed2 (seed2 Long) |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <U> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static CudnnRnnParamsSize <U> create ( Scope escopo, Operando < TInt32 > numLayers, Operando < TInt32 > numUnits, Operando < TInt32 > inputSize, Class <T> T, Class <U> S, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação CudnnRnnParamsSize.
Parâmetros
alcance | escopo atual |
---|---|
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de CudnnRnnParamsSize