|
ComputeNodePtr | syten::STensorImpl::Autodiff::create (std::string &&opname_, Vec< Pair< ComputeNodePtr, Size > > &&input_nodes_, Vec< STensorId > &&output_ids_, AdjointEvaluator &&func_, Vec< STensor > &&output_shapes_, Vec< AsyncCached< STensor > > &&cached_tensors_={}) |
| Helper to create a new ComputeNode and return a shared pointer to it. More...
|
|
ComputeNodePtr | syten::STensorImpl::Autodiff::create_primer (ComputeNodePtr input_node_ptr, Size input_node_number, STensorId output_id, std::int8_t value, SBasis basis_on_tensor, STensor const &shapelike) |
| Helper to create a priming ComputeNode and a return a shared pointer to it. More...
|
|
STensorId | syten::STensorImpl::Autodiff::new_id () |
| Generates a new tensor ID larger than 999 to be used by auto-generated tensors. More...
|
|
STensor | syten::STensorImpl::Autodiff::return_first_output_adjoint (ComputeNode &node, Size input_number, STensorId const result_id) |
| AdjointEvaluator-typed function which simply returns the output adjoint of the first output. More...
|
|
Automatic differentation of STensor
objects infrastructure.