fix output

core/general-server/op/general_response_op.cpp

@@ -106,12 +106,20 @@ int GeneralResponseOp::inference() {
         }
       }
     } else {
+      int var_size = in->at(idx).shape[0];
+      if (var_size == batch_size) {
         for (int j = 0; j < batch_size; ++j) {
           for (int k = j * cap; k < (j + 1) * cap; ++k) {
             res->mutable_insts(j)->mutable_tensor_array(var_idx)->add_data(
                 reinterpret_cast<char *>(&(data_ptr[k])), sizeof(float));
           }
         }
+      } else {
+        for (int j = 0; j < batch_size; ++j) {
+          res->mutable_insts(j)->mutable_tensor_array(var_idx)->add_data(
+              reinterpret_cast<char *>(&(data_ptr[0])), sizeof(float));
+        }
+      }
     }
     var_idx++;
   }