Struct Linear

pub struct Linear<B: Backend> {
    pub weight: Tensor<B>,
    pub bias: Option<Tensor<B>>,
}

Linear layer

Fields

weight: Tensor<B>

weight

bias: Option<Tensor<B>>

bias

Implementations

impl<B: Backend> Linear<B>

pub fn forward(&self, x: impl IntoTensor<B>) -> Tensor<B>

Forward function for linear. Calculates x.dot(&self.weight) + self.bias

Trait Implementations

impl<'a, B: Backend> IntoIterator for &'a Linear<B>

type Item = &'a Tensor<B>

The type of the elements being iterated over.

type IntoIter = IntoIter<&'a Tensor<B>>

Which kind of iterator are we turning this into?

fn into_iter(self) -> Self::IntoIter

Creates an iterator from a value.

impl<'a, B: Backend> IntoIterator for &'a mut Linear<B>

type Item = &'a mut Tensor<B>

The type of the elements being iterated over.

type IntoIter = IntoIter<&'a mut Tensor<B>>

Which kind of iterator are we turning this into?

fn into_iter(self) -> Self::IntoIter

Creates an iterator from a value.