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.
Klasy zagnieżdżone
klasa | CudnnRnnParamsSize.Opcje | Opcjonalne atrybuty dla CudnnRnnParamsSize |
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
Wyjście <U> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
static <U rozszerza TNumber , T rozszerza TNumber > CudnnRnnParamsSize <U> | |
statyczny CudnnRnnParamsSize.Options | kierunek (kierunek struny) |
statyczny CudnnRnnParamsSize.Options | porzucenie (odrzucenie pływające) |
statyczny CudnnRnnParamsSize.Options | inputMode (ciąg wejściowyMode) |
statyczny CudnnRnnParamsSize.Options | numProj (długie numProj) |
Wyjście <U> | |
statyczny CudnnRnnParamsSize.Options | rnnMode (Ciąg rnnMode) |
statyczny CudnnRnnParamsSize.Options | nasiona (długie nasiona) |
statyczny CudnnRnnParamsSize.Options | nasiona 2 (Długie nasiona 2) |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
publiczne wyjście <U> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static CudnnRnnParamsSize <U> create ( Zakres zakresu, Operand < TInt32 > numLayers, Operand < TInt32 > numUnits, Operand < TInt32 > inputSize, Class<T> T, Class<U> S, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CudnnRnnParamsSize.
Parametry
zakres | aktualny zakres |
---|---|
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja CudnnRnnParamsSize