1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

use crate::{ops::{FromVec, Zeros}, tensor::Variable, shape::IntoShape};
use std::cell::RefCell;

/// Initialize tensor from vec and shape
impl<S, T> FromVec<T> for Variable<S>
where
    S: FromVec<T> + Zeros,
{
    fn from_vec(data: Vec<T>, shape: impl IntoShape) -> Self {
        let shape = shape.shape();
        debug_assert_eq!(data.len(), shape.numel());
        Self {
            data: RefCell::new(S::from_vec(data, shape.clone())),
            grad: RefCell::new(S::zeros(shape)),
        }
    }
}