padding on the decoder side

source/sample/transformer/T2TPredictor.cpp

@@ -155,8 +155,12 @@ void T2TPredictor::Predict(T2TStateBundle * next, XTensor * encoding, XTensor *
     XTensor decoding;
     XTensor decodingStep;
     
+    for(int i = 0; i < inputDec.order - 1; i++)
+        dims[i] = inputDec.GetDim(i);
+    dims[inputDec.order - 1] = inputDec.GetDim(-1);
+    
     XTensor paddingDec;
-    InitTensor3D(&paddingDec, inputDec.GetDim(0), inputDec.GetDim(1), m->outputLayer->vSize, X_INT);
+    InitTensor(&paddingDec, inputDec.order, dims, X_INT);
     SetDataFixedInt(paddingDec, 1);
     
     XTensor maskDec;