CudnnRnnParamsSize

Oblicza wielkość wag, które mogą być wykorzystane przez model Cudnn RNN.

Zwróć rozmiar parametrów, który może być używany przez model Cudnn RNN. Późniejsze przypisanie wagi i inicjalizacja powinny wykorzystywać ten rozmiar.

num_layers: Określa liczbę warstw w modelu RNN. num_units: Określa rozmiar stanu ukrytego. input_size: Określa rozmiar stanu wejściowego. rnn_mode: Wskazuje typ modelu RNN. input_mode: Wskazuje, czy istnieje projekcja liniowa pomiędzy danymi wejściowymi a rzeczywistym obliczeniem przed pierwszą warstwą. „skip_input” jest dozwolone tylko wtedy, gdy input_size == num_units; „auto_select” oznacza „skip_input”, gdy input_size == liczba_jednostek; w przeciwnym razie oznacza to „wejście liniowe”. kierunek: wskazuje, czy będzie używany model dwukierunkowy. dir = (kierunek == dwukierunkowy)? Rezygnacja 2:1: prawdopodobieństwo rezygnacji. Po ustawieniu na 0, odrzucanie jest wyłączone. nasiono: pierwsza część materiału siewnego inicjująca porzucenie. nasiono2: druga część materiału siewnego inicjująca porzucenie. params_size: Rozmiar bufora parametrów, który powinien zostać przydzielony i zainicjowany dla tego modelu RNN. Należy pamiętać, że ten bufor parametrów może nie być kompatybilny z różnymi procesorami graficznymi. Użyj CudnnRNNParamsWeights i CudnnRNNParamsBiases, aby zapisać i przywrócić je w sposób kompatybilny w różnych przebiegach.