RFlat2DMatrixOperators.hxx

Go to the documentation of this file.

// Author: Martin Føll, University of Oslo (UiO) & CERN 1/2026
 
/*************************************************************************
 * Copyright (C) 1995-2026, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#ifndef ROOT_INTERNAL_ML_RFLAT2DMATRIXOPERATORS
#define ROOT_INTERNAL_ML_RFLAT2DMATRIXOPERATORS
 
#include <cstddef>
#include <vector>
 
// Forward decl
namespace ROOT::Experimental::Internal::ML {
struct RFlat2DMatrix;
} // namespace ROOT::Experimental::Internal::ML
 
namespace ROOT::Experimental::Internal::ML {
/**
\class ROOT::Experimental::Internal::ML::RFlat2DMatrixOperators
 
\brief Collection of operations applied to one or multiple flat 2D matrices.
*/
 
class RFlat2DMatrixOperators {
private:
   bool fShuffle;
   std::size_t fSetSeed;
 
public:
   RFlat2DMatrixOperators(bool shuffle = true, const std::size_t setSeed = 0) : fShuffle(shuffle), fSetSeed(setSeed) {}
 
   ~RFlat2DMatrixOperators();
 
   // returns the order in which to read the rows of a tensor, empty when shuffling is off
   std::vector<std::size_t> MakePermutation(std::size_t rows);
 
   // shuffles the indices in place, nothing when shuffling is off
   void ShuffleIndices(std::vector<std::size_t> &indices);
 
   // appends the rows [begin, end) of Tensor to Dest, read in the order of the permutation unless it is empty
   static void AppendRows(RFlat2DMatrix &Dest, const RFlat2DMatrix &Tensor, const std::vector<std::size_t> &permutation,
                          std::size_t begin, std::size_t end);
};
 
} // namespace ROOT::Experimental::Internal::ML
#endif // ROOT_INTERNAL_ML_RFLAT2DMATRIXOPERATORS