From 293a7d1e75d14a744852523383bdbef1663887be Mon Sep 17 00:00:00 2001
From: Yang Yang <yangyang62@baidu.com>
Date: Tue, 10 Oct 2017 18:55:16 +0000
Subject: [PATCH] add feed infershape todo

---
 paddle/operators/feed_op.h | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/operators/feed_op.h b/paddle/operators/feed_op.h
index 47344e309c..e406d22209 100644
--- a/paddle/operators/feed_op.h
+++ b/paddle/operators/feed_op.h
@@ -31,6 +31,9 @@ class FeedKernel : public framework::OpKernel<T> {
         g_feed_variable->Get<std::vector<framework::Tensor>>();
     int col = ctx.template Attr<int>("col");
     PADDLE_ENFORCE_GT(tensors.size(), static_cast<size_t>(col));
+    // TODO(qijun):
+    //   check tensors[col].dims() with attribute,
+    //   except the first dimenson.
     out->CopyFrom<T>(tensors[col], ctx.GetPlace());
   }
 };
-- 
GitLab