提交 a0b7a07e 编写于 作者: W wanghaox

fix some comments

...@@ -35,11 +35,14 @@ class SequenceSliceOp : public framework::OperatorWithKernel { ...@@ -35,11 +35,14 @@ class SequenceSliceOp : public framework::OperatorWithKernel {
auto offset_dim = ctx->GetInputDim("Offset"); auto offset_dim = ctx->GetInputDim("Offset");
auto length_dim = ctx->GetInputDim("Length"); auto length_dim = ctx->GetInputDim("Length");
PADDLE_ENFORCE_EQ(offset_dim.size(), 2UL, PADDLE_ENFORCE_EQ(
"Only support one level sequence now."); offset_dim.size(), 2UL,
PADDLE_ENFORCE_EQ(length_dim.size(), 2UL, "Only support one level sequence now, The rank of offset must be 2.");
"Only support one level sequence now."); PADDLE_ENFORCE_EQ(
length_dim.size(), 2UL,
"Only support one level sequence now, The rank of Length must be 2.");
// Initialize the output's dims to maximum
ctx->SetOutputDim("Out", input_dims); ctx->SetOutputDim("Out", input_dims);
} }
......
...@@ -54,10 +54,10 @@ class SequenceSliceOpKernel : public framework::OpKernel<T> { ...@@ -54,10 +54,10 @@ class SequenceSliceOpKernel : public framework::OpKernel<T> {
PADDLE_ENFORCE_EQ(lod.size(), 1UL, PADDLE_ENFORCE_EQ(lod.size(), 1UL,
"Only support one level sequence now."); "Only support one level sequence now.");
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_EQ(
n, length->dims()[0], n, static_cast<size_t>(length->dims()[0]),
"The size of input-sequence and length-array should be the same") "The size of input-sequence and length-array should be the same")
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_EQ(
n, offset->dims()[0], n, static_cast<size_t>(offset->dims()[0]),
"The size of input-sequence and offset-array should be the same") "The size of input-sequence and offset-array should be the same")
const int64_t* offset_data = offset->data<int64_t>(); const int64_t* offset_data = offset->data<int64_t>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册