versions/v1.0.1/Generator_8hpp_source.html

#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_

Device.hpp

Tensor.hpp

Type.hpp

Type
constexpr Type_class Type
data type
Definition Type.hpp:426

cytnx::Tensor
an tensor (multi-dimensional array)
Definition Tensor.hpp:41

cytnx_error.hpp

cytnx
Definition Accessor.hpp:12

cytnx::Device
Device_class Device
data on which devices.

cytnx::linspace
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)

cytnx::arange
Tensor arange(const cytnx_int64 &Nelem)
create an rank-1 Tensor with incremental unsigned integer elements start with [0,Nelem)

cytnx::ones
Tensor ones(const cytnx_uint64 &Nelem, const unsigned int &dtype=Type.Double, const int &device=Device.cpu)
create an rank-1 Tensor with all the elements are initialized with one.

cytnx::zeros
Tensor zeros(const cytnx_uint64 &Nelem, const unsigned int &dtype=Type.Double, const int &device=Device.cpu)
create an rank-1 Tensor with all the elements are initialized with zero.

cytnx::identity
Tensor identity(const cytnx_uint64 &Dim, const unsigned int &dtype=Type.Double, const int &device=Device.cpu)
create an square rank-2 Tensor with all diagonal to be one.

cytnx::eye
Tensor eye(const cytnx_uint64 &Dim, const unsigned int &dtype=Type.Double, const int &device=Device.cpu)
create a square rank-2 Tensor with all diagonal to be one.