rotta_rs 0.1.0

a Deep Learning library with rust language
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

use crate::rotta_rs_module::Tensor;

#[derive(Clone)]
pub struct MSE {}

impl MSE {
    pub fn init() -> Self {
        Self {}
    }

    pub fn forward(&self, prediction: &Tensor, label: &Tensor) -> Tensor {
        (1.0 / (prediction.len() as f32)) * &(label - prediction).powi(2).sum()
    }
}