TensorData.h

Go to the documentation of this file.

// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 
#ifndef EDM4EIC_TensorDATA_H
#define EDM4EIC_TensorDATA_H
 
#include <cstdint>
 
namespace edm4eic {
 
 
/** @class TensorData
 *  Tensor type for use in training in inference of ML models
 *  @author: D. Kalinkin
 */
class TensorData {
public:
  std::int32_t elementType{}; ///< Data type in the same encoding as "ONNXTensorElementDataType", 1 - float, 7 - int64
 
  unsigned int shape_begin{};
  unsigned int shape_end{};
  unsigned int floatData_begin{};
  unsigned int floatData_end{};
  unsigned int int64Data_begin{};
  unsigned int int64Data_end{};
};
 
} // namespace edm4eic
 
 
#endif