From 78fa5e307da3cb32706f396346d3db7a875b4178 Mon Sep 17 00:00:00 2001
From: fengjiayi <fengjiayi@baidu.com>
Date: Mon, 17 Jul 2017 20:00:58 +0800
Subject: [PATCH] Add DDim::size()

---
 paddle/framework/ddim.cc      | 2 ++
 paddle/framework/ddim.h       | 2 ++
 paddle/framework/ddim_test.cc | 1 +
 3 files changed, 5 insertions(+)

diff --git a/paddle/framework/ddim.cc b/paddle/framework/ddim.cc
index 73f5499ad15..b6ad8b60aaf 100644
--- a/paddle/framework/ddim.cc
+++ b/paddle/framework/ddim.cc
@@ -117,6 +117,8 @@ int DDim::operator[](int idx) const {
   return boost::apply_visitor(DynamicConstIndexer(idx), var);
 }
 
+ssize_t DDim::size() const { return arity(*this); }
+
 bool DDim::operator==(DDim d) const {
   if (var.which() != d.getVar().which()) {
     return false;
diff --git a/paddle/framework/ddim.h b/paddle/framework/ddim.h
index a0c2a8a74af..7bc21a1e345 100644
--- a/paddle/framework/ddim.h
+++ b/paddle/framework/ddim.h
@@ -50,6 +50,8 @@ struct DDim {
 
   DDimVar getVar() { return var; }
 
+  ssize_t size() const;
+
   bool operator==(DDim d) const;
 
   bool operator!=(DDim d) const;
diff --git a/paddle/framework/ddim_test.cc b/paddle/framework/ddim_test.cc
index 6a099f2aeb4..9d18a2972ce 100644
--- a/paddle/framework/ddim_test.cc
+++ b/paddle/framework/ddim_test.cc
@@ -49,6 +49,7 @@ TEST(DDim, Equality) {
 
   // arity of a DDim
   EXPECT_EQ(paddle::framework::arity(ddim), 3);
+  EXPECT_EQ(ddim.size(), 3);
 
   // product of a DDim
   EXPECT_EQ(paddle::framework::product(vddim), 45);
-- 
GitLab