diff --git a/source/RnnStack.hpp b/source/RnnStack.hpp
index 9a1eeeb..a9751df 100644
--- a/source/RnnStack.hpp
+++ b/source/RnnStack.hpp
@@ -15,6 +15,7 @@
 #define RNNSTACK_HPP
 
 #include "AStack.hpp"
+#include "matutils.hpp"
 
 class RnnStack : public AStack
 {
@@ -33,6 +34,27 @@ public:
 	
 	arma::mat step_forward(arma::mat &state, const arma::mat &v);
 
+	void sample_one_hot(arma::mat &srcDst)
+	{
+		double k = arma::accu(srcDst);
+		
+		if (k != 0)
+		{
+			srcDst = srcDst / k;
+		}
+		
+		arma::mat ps = Matutils::sample(srcDst);
+		if (arma::accu(ps) > 1)
+		{
+			arma::uvec q1 = find(ps > 0);
+			int winner = (q1.n_elem-1) * arma::randu(1)[0];
+			int wix = q1(winner);
+			
+			srcDst = zeros(arma::size(srcDst));
+			srcDst[wix] = 1;
+		}
+	}
+	
 	void clamp_one_hot(arma::mat &srcDst)
 	{
 		int index = srcDst.index_max();