Commit 63f0f6a7 by xiaotong

back to the model that an XTensor can not be assigned when constructed

parent a1835883
...@@ -76,9 +76,12 @@ search for the most promising states ...@@ -76,9 +76,12 @@ search for the most promising states
*/ */
void T2TSearch::Search(T2TModel * model, XTensor * input, XTensor * padding, XTensor * output) void T2TSearch::Search(T2TModel * model, XTensor * input, XTensor * padding, XTensor * output)
{ {
XTensor maskEnc;
XTensor encoding;
T2TPredictor predictor; T2TPredictor predictor;
XTensor maskEnc;
XTensor encoding;
XTensor encodingBeam;
XTensor inputBeam;
XTensor paddingBeam;
CheckNTErrors(endSymbolNum > 0, "The search class is not initialized!"); CheckNTErrors(endSymbolNum > 0, "The search class is not initialized!");
CheckNTErrors(startSymbol >= 0, "The search class is not initialized!"); CheckNTErrors(startSymbol >= 0, "The search class is not initialized!");
...@@ -95,9 +98,9 @@ void T2TSearch::Search(T2TModel * model, XTensor * input, XTensor * padding, XTe ...@@ -95,9 +98,9 @@ void T2TSearch::Search(T2TModel * model, XTensor * input, XTensor * padding, XTe
encoding = model->MakeEncoder(*input, maskEnc, false); encoding = model->MakeEncoder(*input, maskEnc, false);
encoding.SetName(ENCODING_NAME); encoding.SetName(ENCODING_NAME);
XTensor encodingBeam = Unsqueeze(encoding, encoding.order - 2, beamSize); encodingBeam = Unsqueeze(encoding, encoding.order - 2, beamSize);
XTensor inputBeam = Unsqueeze(*input, input->order - 1, beamSize); inputBeam = Unsqueeze(*input, input->order - 1, beamSize);
XTensor paddingBeam = Unsqueeze(*padding, padding->order - 1, beamSize); paddingBeam = Unsqueeze(*padding, padding->order - 1, beamSize);
encodingBeam.ReshapeMerged(encodingBeam.order - 4); encodingBeam.ReshapeMerged(encodingBeam.order - 4);
inputBeam.ReshapeMerged(inputBeam.order - 3); inputBeam.ReshapeMerged(inputBeam.order - 3);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论