TensorOptimizationAlgorithm

Trait TensorOptimizationAlgorithm 

Source
pub trait TensorOptimizationAlgorithm:
    Send
    + Sync
    + Debug {
    // Required methods
    fn optimize(
        &self,
        network: &mut TensorNetwork,
        target: &Tensor,
    ) -> Result<OptimizationResult, TensorNetworkError>;
    fn get_algorithm_name(&self) -> &str;
    fn get_parameters(&self) -> HashMap<String, f64>;
}
Expand description

Tensor optimization algorithm trait

Required Methods§

Source

fn optimize( &self, network: &mut TensorNetwork, target: &Tensor, ) -> Result<OptimizationResult, TensorNetworkError>

Optimize tensor network

Source

fn get_algorithm_name(&self) -> &str

Get algorithm name

Source

fn get_parameters(&self) -> HashMap<String, f64>

Get algorithm parameters

Implementors§