cubeai_concepts.h

Go to the documentation of this file.

#ifndef CUBEAI_CONCEPTS_H
#define CUBEAI_CONCEPTS_H
#include "cuberl/base/cubeai_config.h"
 
#ifdef USE_PYTORCH
#include "torch/torch.h"
#endif
 
 
#include <type_traits>
 
namespace cuberl{
namespace utils {
namespace concepts {
 
template<typename VectorType>
concept integral_vector = std::is_integral<typename VectorType::value_type>::value;
 
template<typename VectorType>
concept float_vector = std::is_floating_point<typename VectorType::value_type>::value;
 
template<typename VectorType>
concept float_or_integral_vector = std::is_integral<typename VectorType::value_type>::value || std::is_floating_point<typename VectorType::value_type>::value;
 
template<typename Type>
concept is_default_constructible = std::is_default_constructible<Type>::value;
 
 
#ifdef USE_PYTORCH
template<typename AlgoType>
concept pytorch_module = std::is_base_of<torch::nn::Module, AlgoType>::value;
#endif
 
}
 
}
}
 
#endif // CUBEAI_CONSTRAINTS_H