Adding Variable::dims()

pavanky · umar456 · commit 4ad39d844d15 · 2017-07-26T23:29:24.000-04:00
diff --git a/include/af/autograd/Variable.hpp b/include/af/autograd/Variable.hpp
@@ -60,6 +60,8 @@ namespace af {
 
             bool isGradAvailable() const;
 
+            af::dim4 dims() const;
+
             void zeroGrad();
 
             void setCalcGrad(bool calc_grad);
diff --git a/src/autograd/Variable.cpp b/src/autograd/Variable.cpp
@@ -102,6 +102,11 @@ namespace af {
             return m_shared->m_grads.size() >= 1;
         }
 
+        af::dim4 Variable::dims() const
+        {
+            return m_shared->m_data.dims();
+        }
+
         void Variable::zeroGrad()
         {
             m_shared->m_grads.clear();
