Imported from https://github.com/alexmlamb/blocks_rl_gru_setup