#include <list>#include <limits>#include <mlpack/prereqs.hpp>#include "../visitor/delta_visitor.hpp"#include "../visitor/output_parameter_visitor.hpp"#include "layer_types.hpp"#include "add_merge.hpp"#include "sequential.hpp"#include "gru_impl.hpp"
Include dependency graph for gru.hpp:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
Classes | |
| class | mlpack::ann::GRU< InputDataType, OutputDataType > |
| An implementation of a gru network layer. More... | |
Namespaces | |
| mlpack | |
| Linear algebra utility functions, generally performed on matrices or vectors. | |
| mlpack::ann | |
| Artificial Neural Network. | |
Detailed Description
Definition of the GRU layer.
For more information, read the following paper:
@inproceedings{chung2015gated,
title = {Gated Feedback Recurrent Neural Networks.},
author = {Chung, Junyoung and G{\"u}l{\c{c}}ehre, Caglar and Cho,
Kyunghyun and Bengio, Yoshua},
booktitle = {ICML},
pages = {2067--2075},
year = {2015},
url = {https://arxiv.org/abs/1502.02367}
}
mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.
