Fix infer shape of warpctc op. (#20656)

wanghaoshuang · web-flow · commit 06e35ae42495 · 2019-10-16T20:45:50.000+08:00
diff --git a/paddle/fluid/operators/warpctc_op.cc b/paddle/fluid/operators/warpctc_op.cc
@@ -52,7 +52,7 @@ class WarpCTCOp : public framework::OperatorWithKernel {
                    sequence_width);
 
     // TODO(liuyiqun): it is tricky to set the wrong dimension here.
-    ctx->SetOutputDim("Loss", {logits_dims[0], 1});
+    ctx->SetOutputDim("Loss", {-1, 1});
   }
 
  protected:

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@ class WarpCTCOp : public framework::OperatorWithKernel {`
`52`	`52`	`sequence_width);`
`53`	`53`
`54`	`54`	`// TODO(liuyiqun): it is tricky to set the wrong dimension here.`
`55`		`- ctx->SetOutputDim("Loss", {logits_dims[0], 1});`
	`55`	`+ ctx->SetOutputDim("Loss", {-1, 1});`
`56`	`56`	`}`
`57`	`57`
`58`	`58`	`protected:`