Generator.hpp

Go to the documentation of this file.

#ifndef CYTNX_GENERATOR_H_
#define CYTNX_GENERATOR_H_
 
#include "Type.hpp"
#include "Device.hpp"
#include "cytnx_error.hpp"
#include "Tensor.hpp"
#include <vector>
#include <initializer_list>
 
#ifdef BACKEND_TORCH
#else
namespace cytnx {
 
 
  Tensor zeros(const cytnx_uint64 &Nelem, const unsigned int &dtype = Type.Double,
               const int &device = Device.cpu);
  Tensor zeros(const std::vector<cytnx_uint64> &Nelem, const unsigned int &dtype = Type.Double,
               const int &device = Device.cpu);
 
  Tensor identity(const cytnx_uint64 &Dim, const unsigned int &dtype = Type.Double,
                  const int &device = Device.cpu);
 
  Tensor eye(const cytnx_uint64 &Dim, const unsigned int &dtype = Type.Double,
             const int &device = Device.cpu);
 
 
  Tensor ones(const cytnx_uint64 &Nelem, const unsigned int &dtype = Type.Double,
              const int &device = Device.cpu);
  Tensor ones(const std::vector<cytnx_uint64> &Nelem, const unsigned int &dtype = Type.Double,
              const int &device = Device.cpu);
 
 
  Tensor arange(const cytnx_int64 &Nelem);
  Tensor arange(const cytnx_double &start, const cytnx_double &end, const cytnx_double &step = 1,
                const unsigned int &dtype = Type.Double, const int &device = Device.cpu);
 
  Tensor linspace(const cytnx_double &start, const cytnx_double &end, const cytnx_uint64 &Nelem,
                  const bool &endpoint = true, const unsigned int &dtype = Type.Double,
                  const int &device = Device.cpu);
 
  // Tensor rand(const std::vector<cytnx_uint64> &size);
  // Tensor randn(const std::vector<cytnx_uint64> &size);
  // Tensor normal(const cytnx_double &mean, cytnx_double &std);
 
}  // namespace cytnx
 
#endif  // BACKEND_TORCH
 
#endif  // CYTNX_GENERATOR_H_