lumen-core 0.6.1

A tiny ML framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

use crate::{Tensor, WithDType};

pub enum TensorOrScalar<T: WithDType> {
    Tensor(Tensor<T>),
    Scalar(T),
}

impl<T: WithDType> From<T> for TensorOrScalar<T> {
    fn from(value: T) -> Self {
        TensorOrScalar::Scalar(value)
    }
}

impl<T: WithDType> From<Tensor<T>> for TensorOrScalar<T> {
    fn from(value: Tensor<T>) -> Self {
        TensorOrScalar::Tensor(value)
    }
}

impl<T: WithDType> From<&Tensor<T>> for TensorOrScalar<T> {
    fn from(value: &Tensor<T>) -> Self {
        TensorOrScalar::Tensor(value.clone())
    }
}