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 std::vector<cytnx_uint64> &shape, unsigned int dtype = Type.Double,
               int device = Device.cpu);
  Tensor zeros(std::initializer_list<cytnx_uint64> shape, unsigned int dtype = Type.Double,
               int device = Device.cpu);
 
  Tensor identity(cytnx_uint64 Dim, unsigned int dtype = Type.Double, int device = Device.cpu);
 
  Tensor eye(cytnx_uint64 Dim, unsigned int dtype = Type.Double, int device = Device.cpu);
 
 
  Tensor ones(const std::vector<cytnx_uint64> &shape, unsigned int dtype = Type.Double,
              int device = Device.cpu);
  Tensor ones(std::initializer_list<cytnx_uint64> shape, unsigned int dtype = Type.Double,
              int device = Device.cpu);
 
 
  Tensor arange(cytnx_int64 Nelem);
  Tensor arange(cytnx_double start, cytnx_double end, cytnx_double step = 1,
                unsigned int dtype = Type.Double, int device = Device.cpu);
 
  Tensor linspace(cytnx_double start, cytnx_double end, cytnx_uint64 Nelem, bool endpoint = true,
                  unsigned int dtype = Type.Double, 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_