Trait CustomLayer 

pub trait CustomLayer: Send + Sync {
    // Required methods
    fn name(&self) -> &str;
    fn layer_type(&self) -> &str;
    fn input_shape(&self) -> Vec<i64>;
    fn output_shape(&self, input_shape: &[i64]) -> Vec<i64>;
    fn initialize_parameters(&self) -> Result<HashMap<String, TensorRef>>;
    fn config(&self) -> Value;
}

Custom layer interface for advanced builders

Required Methods

fn name(&self) -> &str

Get layer name

fn layer_type(&self) -> &str

Get layer type

fn input_shape(&self) -> Vec<i64>

Get input shape requirements

fn output_shape(&self, input_shape: &[i64]) -> Vec<i64>

Get output shape

fn initialize_parameters(&self) -> Result<HashMap<String, TensorRef>>

Initialize layer parameters

fn config(&self) -> Value

Get layer configuration

Implementors