init_gate_bias: Initialize gates to pass full information

View source: R/initialization.R

init_gate_biasR Documentation

Initialize gates to pass full information

Description

x <- list(rnn_layer = nn_lstm(2, 20)) init_gate_bias(x$rnn_layer) x$rnn_layer$parameters$bias_ih_l1

Usage

init_gate_bias(rnn_layer)

Details

~LSTM.bias_ih_lk – (b_ii|b_if|b_ig|b_io), of shape (4hidden_size) ~LSTM.bias_hh_lk – (b_hi|b_hf|b_hg|b_ho), of shape (4hidden_size)


krzjoa/torchts documentation built on June 24, 2022, 5:30 a.m.