Reduce base margin to 2 dim for now. (#7455)

2021-11-27 00:46:13 +08:00
parent bf7bb575b4
commit 557ffc4bf5
7 changed files with 33 additions and 33 deletions
--- a/src/data/simple_dmatrix.cc
+++ b/src/data/simple_dmatrix.cc
@@ -137,10 +137,10 @@ SimpleDMatrix::SimpleDMatrix(AdapterT* adapter, float missing, int nthread) {
                     batch.Weights() + batch.Size());
    }
    if (batch.BaseMargin() != nullptr) {
-      info_.base_margin_ = linalg::Tensor<float, 3>{batch.BaseMargin(),
-                                                    batch.BaseMargin() + batch.Size(),
-                                                    {batch.Size()},
-                                                    GenericParameter::kCpuId};
+      info_.base_margin_ = decltype(info_.base_margin_){batch.BaseMargin(),
+                                                        batch.BaseMargin() + batch.Size(),
+                                                        {batch.Size()},
+                                                        GenericParameter::kCpuId};
    }
    if (batch.Qid() != nullptr) {
      qids.insert(qids.end(), batch.Qid(), batch.Qid() + batch.Size());