From d9bc6c8f307ed57adfbba2ff450c848bb0c546ef Mon Sep 17 00:00:00 2001
From: abh2k <41710346+abh2k@users.noreply.github.com>
Date: Tue, 15 Jun 2021 10:25:56 +0530
Subject: [PATCH] added author link

---
 src/mlpack/methods/ann/layer/channel_shuffle.hpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/mlpack/methods/ann/layer/channel_shuffle.hpp b/src/mlpack/methods/ann/layer/channel_shuffle.hpp
index 27c747e84a..5e3eb273cb 100644
--- a/src/mlpack/methods/ann/layer/channel_shuffle.hpp
+++ b/src/mlpack/methods/ann/layer/channel_shuffle.hpp
@@ -21,6 +21,19 @@ namespace ann /** Artificial Neural Network. */ {
  * Channel Shuffle divides the channels/units in a tensor into groups
  * and rearrange while keeping the original tensor shape.
  *
+ * For more information, refer to the following paper,
+ *
+ * @code
+ * @article{zhang2018shufflenet,
+ *   author    = {Xiangyu Zhang, Xinyu Zhou, Mengxiao Lin, Jian Sun and
+ *                Megvii Inc},
+ *   title     = {Shufflenet: An extremely efficient convolutional neural
+ *                network for mobile devices},
+ *   year      = {2018},
+ *   url       = {https://arxiv.org/pdf/1707.01083},
+ * }
+ * @endcode
+ *
  * @tparam InputDataType Type of the input data (arma::colvec, arma::mat,
  *         arma::sp_mat or arma::cube).
  * @tparam OutputDataType Type of the output data (arma::colvec, arma::mat,