Fix infer shape of warpctc op. (#20656)

06e35ae4 · whs · GitHub · 7b9b7303 · 06e35ae4
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/warpctc_op.cc paddle/fluid/operators/warpctc_op.cc +1 -1

未找到文件。
--- 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: