Cyclops Tensor Framework
parallel arithmetic on multidimensional arrays
ctf_ext.h
Go to the documentation of this file.
1 
2 #include "../include/ctf.hpp"
3 
4 namespace CTF_int{
5 
6 
12  template <typename dtype>
13  void abs_helper(tensor * A, tensor * B);
14 
20  template <typename dtype>
21  void pow_helper(tensor * A, tensor * B, tensor * C, char const * idx_A, char const * idx_B, char const * idx_C);
22 
28  template <typename dtype>
29  void all_helper(tensor * A, tensor * B_bool, char const * idx_A, char const * idx_B);
30 
31  template <typename dtype>
32  void conj_helper(tensor * A, tensor * B);
33 
39  template <typename dtype>
40  void get_real(tensor * A, tensor * B);
41 
47  template <typename dtype>
48  void get_imag(tensor * A, tensor * B);
49 
55  template <typename dtype>
56  void set_real(tensor * A, tensor * B);
57 
63  template <typename dtype>
64  void set_imag(tensor * A, tensor * B);
65 
71  template <typename dtype>
72  void any_helper(tensor * A, tensor * B_bool, char const * idx_A, char const * idx_B);
78  int64_t sum_bool_tsr(tensor * A);
79 
85  void subsample(tensor * A, double probability);
86 
87  void matrix_svd(tensor * A, tensor * U, tensor * S, tensor * VT, int rank);
88  void matrix_svd_cmplx(tensor * A, tensor * U, tensor * S, tensor * VT, int rank);
89 
90  void matrix_qr(tensor * A, tensor * Q, tensor * R);
91  void matrix_qr_cmplx(tensor * A, tensor * Q, tensor * R);
92 
100  void conv_type(int type_idx1, int type_idx2, tensor * A, tensor * B);
101 
102 }
ctf.core.rank
def rank(self)
Definition: core.pyx:312
CTF_int::sum_bool_tsr
int64_t sum_bool_tsr(tensor *A)
sum all 1 values in boolean tensor
Definition: ctf_ext.cxx:92
CTF_int::matrix_svd
void matrix_svd(tensor *A, tensor *U, tensor *S, tensor *VT, int rank)
Definition: ctf_ext.cxx:175
CTF_int::conv_type
void conv_type(int type_idx1, int type_idx2, tensor *A, tensor *B)
convert tensor from one type to another
Definition: ctf_ext.cxx:327
CTF_int::get_real
void get_real(tensor *A, tensor *B)
Definition: ctf_ext.cxx:51
CTF_int::abs_helper
void abs_helper(tensor *A, tensor *B)
Definition: ctf_ext.cxx:17
CTF_int::all_helper
void all_helper(tensor *A, tensor *B_bool, char const *idx_A, char const *idx_B)
Definition: ctf_ext.cxx:33
CTF_int::matrix_qr_cmplx
void matrix_qr_cmplx(tensor *A, tensor *Q, tensor *R)
Definition: ctf_ext.cxx:141
CTF_int::subsample
void subsample(tensor *A, double probability)
extract a sample of the entries (if sparse of the current nonzeros)
Definition: ctf_ext.cxx:103
CTF_int::get_imag
void get_imag(tensor *A, tensor *B)
Definition: ctf_ext.cxx:60
CTF_int::set_real
void set_real(tensor *A, tensor *B)
Definition: ctf_ext.cxx:70
CTF_int::conj_helper
void conj_helper(tensor *A, tensor *B)
Definition: ctf_ext.cxx:42
CTF_int::any_helper
void any_helper(tensor *A, tensor *B_bool, char const *idx_A, char const *idx_B)
Definition: ctf_ext.cxx:88
CTF_int::pow_helper
void pow_helper(tensor *A, tensor *B, tensor *C, char const *idx_A, char const *idx_B, char const *idx_C)
Definition: ctf_ext.cxx:26
CTF_int::matrix_qr
void matrix_qr(tensor *A, tensor *Q, tensor *R)
Definition: ctf_ext.cxx:109
CTF_int::set_imag
void set_imag(tensor *A, tensor *B)
Definition: ctf_ext.cxx:79
CTF_int
Definition: model_trainer.cxx:16
CTF_int::matrix_svd_cmplx
void matrix_svd_cmplx(tensor *A, tensor *U, tensor *S, tensor *VT, int rank)
Definition: ctf_ext.cxx:213