![]() |
KIDS
ver-0.0.1
KIDS : Kernel Integrated Dynamics Simulator
|
#include <Einsum.h>
Public Member Functions | |
EinsumHelper (const std::string &einsum_expression, std::vector< std::vector< std::size_t > > shape_inputs, std::vector< std::size_t > shape_output={}) | |
Public Attributes | |
std::size_t | total_esidx |
total number of EinsumIdx in EinsumIdx System | |
std::size_t | total_tensor |
total number of tensor in einsum rule | |
std::vector< EinsumIdx > | einsum_idxs |
the EinsumIdx System | |
std::vector< std::size_t > | einsum_dims |
each dimension of EinsumIdx System | |
std::vector< std::string > | fixed_label_names |
store for fixed labels | |
std::vector< std::string > | esshape_inputs |
store einsum's strings of input tensors | |
std::string | esshape_output = "" |
store/deduct einsum's for the ouput tensor | |
std::vector< DimenHelper > | dh_inputs |
DimenHelper for input tensors. | |
DimenHelper | dh_output |
DimenHelper for ouput tensor. | |
std::vector< std::size_t > | einsum_iposes |
idx placeholder for EinsumIdx System | |
std::vector< std::size_t > | ipos_inputs |
idx placeholder for input tensors | |
int | count1 = 0 |
int | count2 = 0 |
int | count3 = 0 |
int | total_loop = 0 |
PROJECT_NS::EinsumHelper::EinsumHelper | ( | const std::string & | einsum_expression, |
std::vector< std::vector< std::size_t > > | shape_inputs, | ||
std::vector< std::size_t > | shape_output = {} ) |
[in] | einsum_expression | expression for einsum rule |
[in] | shape_inputs | input shapes as a vector |
[in] | shape_output | output shapes |
Definition at line 43 of file Einsum.cpp.
References count1, count2, count3, dh_inputs, dh_output, einsum_dims, einsum_idxs, einsum_iposes, esshape_inputs, esshape_output, fixed_label_names, ipos_inputs, total_esidx, total_loop, and total_tensor.
int PROJECT_NS::EinsumHelper::count1 = 0 |
Definition at line 196 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
int PROJECT_NS::EinsumHelper::count2 = 0 |
Definition at line 197 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), and EinsumHelper().
int PROJECT_NS::EinsumHelper::count3 = 0 |
Definition at line 198 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
std::vector<DimenHelper> PROJECT_NS::EinsumHelper::dh_inputs |
DimenHelper for input tensors.
Definition at line 190 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
DimenHelper PROJECT_NS::EinsumHelper::dh_output |
DimenHelper for ouput tensor.
Definition at line 191 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
std::vector<std::size_t> PROJECT_NS::EinsumHelper::einsum_dims |
each dimension of EinsumIdx System
Definition at line 183 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
std::vector<EinsumIdx> PROJECT_NS::EinsumHelper::einsum_idxs |
std::vector<std::size_t> PROJECT_NS::EinsumHelper::einsum_iposes |
idx placeholder for EinsumIdx System
Definition at line 193 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
std::vector<std::string> PROJECT_NS::EinsumHelper::esshape_inputs |
store einsum's strings of input tensors
Definition at line 187 of file Einsum.h.
Referenced by EinsumHelper().
std::string PROJECT_NS::EinsumHelper::esshape_output = "" |
store/deduct einsum's for the ouput tensor
Definition at line 188 of file Einsum.h.
Referenced by EinsumHelper().
std::vector<std::string> PROJECT_NS::EinsumHelper::fixed_label_names |
std::vector<std::size_t> PROJECT_NS::EinsumHelper::ipos_inputs |
idx placeholder for input tensors
Definition at line 194 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
std::size_t PROJECT_NS::EinsumHelper::total_esidx |
total number of EinsumIdx in EinsumIdx System
Definition at line 179 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
int PROJECT_NS::EinsumHelper::total_loop = 0 |
Definition at line 199 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().
std::size_t PROJECT_NS::EinsumHelper::total_tensor |
total number of tensor in einsum rule
Definition at line 180 of file Einsum.h.
Referenced by PROJECT_NS::einsum(), PROJECT_NS::einsum_fun(), and EinsumHelper().