random.hpp

Go to the documentation of this file.

#ifndef _random_H_
#define _random_H_
#include "Type.hpp"
#include "Device.hpp"
#include "cytnx_error.hpp"
#include "Tensor.hpp"
#include "Storage.hpp"
#include "UniTensor.hpp"
#include <vector>
#include <initializer_list>
#include <random>
 
namespace cytnx {
  namespace random {
 
    // [Storage]
    // =============================================================================================================
    // =============================================================================================================
    // Make_normal:
    //--------------------------------------------------
    void Make_normal(Storage &Sin, const double &mean, const double &std,
                     const unsigned int &seed = std::random_device()());
 
    // Make_uniform:
    //--------------------------------------------------
    void Make_uniform(Storage &Sin, const double &low = 0, const double &high = 1,
                      const unsigned int &seed = std::random_device()());
 
    // [Tensor]
    // =============================================================================================================
    // =============================================================================================================
    // Make_normal:
    //-------------------------------------------------
    void Make_normal(Tensor &Tin, const double &mean, const double &std,
                     const unsigned int &seed = std::random_device()());
 
    // Make_uniform:
    //--------------------------------------------------
    void Make_uniform(Tensor &Tin, const double &low = 0, const double &high = 1,
                      const unsigned int &seed = std::random_device()());
 
    // [UniTensor]
    // =============================================================================================================
    // =============================================================================================================
    // Make_normal:
    //-------------------------------------------------
    void Make_normal(UniTensor &Tin, const double &mean, const double &std,
                     const unsigned int &seed = std::random_device()());
 
    // Make_uniform:
    //--------------------------------------------------
    void Make_uniform(UniTensor &Tin, const double &low = 0, const double &high = 1,
                      const unsigned int &seed = std::random_device()());
 
    // normal:
 
    Tensor normal(const cytnx_uint64 &Nelem, const double &mean, const double &std,
                  const int &device = Device.cpu, const unsigned int &seed = std::random_device()(),
                  const unsigned int &dtype = Type.Double);
 
    Tensor normal(const std::vector<cytnx_uint64> &Nelem, const double &mean, const double &std,
                  const int &device = Device.cpu, const unsigned int &seed = std::random_device()(),
                  const unsigned int &dtype = Type.Double);
 
    // uniform:
 
    Tensor uniform(const cytnx_uint64 &Nelem, const double &low, const double &high,
                   const int &device = Device.cpu,
                   const unsigned int &seed = std::random_device()(),
                   const unsigned int &dtype = Type.Double);
 
    Tensor uniform(const std::vector<cytnx_uint64> &Nelem, const double &low, const double &high,
                   const int &device = Device.cpu,
                   const unsigned int &seed = std::random_device()(),
                   const unsigned int &dtype = Type.Double);
 
  }  // namespace random
}  // namespace cytnx
 
#endif