Struct SoftmaxLossLayer 

pub struct SoftmaxLossLayer {
    pub size: usize,
    pub lr_params: CpuParams,
    /* private fields */
}

Fields

size: usize

lr_params: CpuParams

Implementations

impl SoftmaxLossLayer

pub fn new(size: usize) -> Self

pub fn new_box(size: usize) -> Box<Self>

Trait Implementations

impl AbstractLayer for SoftmaxLossLayer

fn forward(&mut self, input: ParamsBlob) -> LayerForwardResult

fn backward_output( &mut self, prev_input: ParamsBlob, expected_vec: Array2D, ) -> LayerBackwardResult

fn layer_type(&self) -> &str

fn cpu_params(&self) -> Option<CpuParams>

fn set_cpu_params(&mut self, lp: CpuParams)

fn size(&self) -> usize

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

fn trainable_bufs(&self) -> TrainableBufsIds<'_>

fn serializable_bufs(&self) -> &[i32]

fn copy_layer(&self) -> Box<dyn AbstractLayer>

fn clone_layer(&self) -> Box<dyn AbstractLayer>

fn set_input_shape(&mut self, sh: &[usize])

fn forward_input(&mut self, _input_data: Array2D) -> LayerForwardResult

fn backward( &mut self, _prev_input: ParamsBlob, _input: ParamsBlob, ) -> LayerBackwardResult

returns out_values and array of weights

fn set_batch_size(&mut self, batch_size: usize)

impl Clone for SoftmaxLossLayer

fn clone(&self) -> SoftmaxLossLayer

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for SoftmaxLossLayer

fn default() -> SoftmaxLossLayer

Returns the “default value” for a type.

impl WithParams for SoftmaxLossLayer

fn cfg(&self) -> HashMap<String, Variant>

fn set_cfg(&mut self, cfg: &HashMap<String, Variant>)