add feed infershape todo

revert-4814-Add_sequence_project_op
Yang Yang 8 years ago
parent 3f9e247a73
commit 293a7d1e75

@ -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());
}
};

Loading…
Cancel
Save