Struct burn::nn::Linear

source · [−]

pub struct Linear<B: Backend> { /* private fields */ }

Expand description

Applies a linear transformation to the input tensor:

O = IW + b

Implementations

source

impl<B: Backend> Linear

source

pub fn new(config: &LinearConfig) -> Self

Trait Implementations

source

impl<B: Backend> ADModule for Linearwhere
B: ADBackend,

type ADBackend = B

type InnerModule = Linear<::InnerBackend>

source

fn inner(&self) -> Self::InnerModule

Get the same module, but on the inner backend without auto-differentiation.

source

impl<B: Debug + Backend> Debug for Linear

source

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

source

impl<B: Backend> Display for Linear

source

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

source

impl<B: Backend, const D: usize> Forward<Tensor<B, D>, Tensor<B, D>> for Linear

source

fn forward(&self, input: Tensor<B, D>) -> Tensor<B, D>

The forward method mapping the input to the output.

source

impl<B: Backend> Module for Linear

type Backend = B

source

fn devices(&self) -> Vec<B::Device>

Get the device list of the module and all of its sub-modules.

source

fn to_device(&mut self, device: B::Device)

Move the module and all of its sub-modules to the given device.

source

fn state(&self) -> State<<Self::Backend as Backend>::Elem>

Get the module state.

source

fn load(
&mut self,
state: &State<<Self::Backend as Backend>::Elem>
) -> Result<(), LoadingError>

Load the module state.

source

fn num_params(&self) -> usize

Get the number of parameters the module has, including all of its sub-modules.

source

fn update_params<O: Optimizer<Backend = B>>(
 &mut self,
 grads: &Gradients,
 optim: &mut O
)where
 B: ADBackend,

Update the module parameters with the given gradients and optimizer.

source

fn load_optim_state<O: Optimizer<Backend = B>>(
 &self,
 optim: &mut O,
 state_optim: &StateNamed<B::Elem>
)where
 B: ADBackend,

Load the optimizer state for the module, including all of its sub-modules. Read more

source

fn register_optim_state<O: Optimizer<Backend = B>>(
 &self,
 optim: &O,
 state_optim: &mut StateNamed<B::Elem>
)where
 B: ADBackend,

Register the optimizer state for the module, including all of its sub-modules. Read more

Auto Trait Implementations

impl RefUnwindSafe for Linearwhere
::TensorPrimitive<1>: RefUnwindSafe,
::TensorPrimitive<2>: RefUnwindSafe,

impl Send for Linear

impl Sync for Linear

impl Unpin for Linearwhere
::TensorPrimitive<1>: Unpin,
::TensorPrimitive<2>: Unpin,

impl UnwindSafe for Linearwhere
::TensorPrimitive<1>: UnwindSafe,
::TensorPrimitive<2>: UnwindSafe,

Blanket Implementations

source

impl<T> Any for Twhere
T: 'static + ?Sized,

source

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

source

impl<T> Borrow<T> for Twhere
T: ?Sized,

const: unstable · source

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

source

impl<T> BorrowMut<T> for Twhere
T: ?Sized,

const: unstable · source

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> Fake for T

fn fake(&self) -> Uwhere
Self: FakeBase,

fn fake_with_rng<U, R>(&self, rng: &mut R) -> Uwhere
R: Rng + ?Sized,
Self: FakeBase,

source

impl<T> From<T> for T

const: unstable · source

fn from(t: T) -> T

Returns the argument unchanged.

source

impl<T, U> Into for Twhere
U: From<T>,

const: unstable · source

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

source

impl<T> Same<T> for T

type Output = T

Should always be Self

source

impl<T> ToString for Twhere
T: Display + ?Sized,

source

default fn to_string(&self) -> String

Converts the given value to a String. Read more

source

impl<T, U> TryFrom for Twhere
U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

const: unstable · source

fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>

Performs the conversion.

source

impl<T, U> TryInto for Twhere
U: TryFrom<T>,

type Error = >::Error

The type returned in the event of a conversion error.

const: unstable · source

fn try_into(self) -> Result<U, >::Error>

Performs the conversion.

Struct burn::nn::Linear

Implementations

impl<B: Backend> Linear<B>

pub fn new(config: &LinearConfig) -> Self

Trait Implementations

impl<B: Backend> ADModule for Linear<B>where B: ADBackend,

type ADBackend = B

type InnerModule = Linear<<B as ADBackend>::InnerBackend>

fn inner(&self) -> Self::InnerModule

impl<B: Debug + Backend> Debug for Linear<B>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

impl<B: Backend> Display for Linear<B>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

impl<B: Backend, const D: usize> Forward<Tensor<B, D>, Tensor<B, D>> for Linear<B>

fn forward(&self, input: Tensor<B, D>) -> Tensor<B, D>

impl<B: Backend> Module for Linear<B>

type Backend = B

fn devices(&self) -> Vec<B::Device>

fn to_device(&mut self, device: B::Device)

fn state(&self) -> State<<Self::Backend as Backend>::Elem>

fn load( &mut self, state: &State<<Self::Backend as Backend>::Elem>) -> Result<(), LoadingError>

fn num_params(&self) -> usize

fn update_params<O: Optimizer<Backend = B>>( &mut self, grads: &Gradients, optim: &mut O)where B: ADBackend,

fn load_optim_state<O: Optimizer<Backend = B>>( &self, optim: &mut O, state_optim: &StateNamed<B::Elem>)where B: ADBackend,

fn register_optim_state<O: Optimizer<Backend = B>>( &self, optim: &O, state_optim: &mut StateNamed<B::Elem>)where B: ADBackend,

Auto Trait Implementations

impl<B> RefUnwindSafe for Linear<B>where <B as Backend>::TensorPrimitive<1>: RefUnwindSafe, <B as Backend>::TensorPrimitive<2>: RefUnwindSafe,

impl<B> Send for Linear<B>

impl<B> Sync for Linear<B>

impl<B> Unpin for Linear<B>where <B as Backend>::TensorPrimitive<1>: Unpin, <B as Backend>::TensorPrimitive<2>: Unpin,

impl<B> UnwindSafe for Linear<B>where <B as Backend>::TensorPrimitive<1>: UnwindSafe, <B as Backend>::TensorPrimitive<2>: UnwindSafe,

Blanket Implementations

impl<T> Any for Twhere T: 'static + ?Sized,

fn type_id(&self) -> TypeId

impl<T> Borrow<T> for Twhere T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for Twhere T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<T> Fake for T

fn fake<U>(&self) -> Uwhere Self: FakeBase<U>,

fn fake_with_rng<U, R>(&self, rng: &mut R) -> Uwhere R: Rng + ?Sized, Self: FakeBase<U>,

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Into<U> for Twhere U: From<T>,

fn into(self) -> U

impl<T> Same<T> for T

type Output = T

impl<T> ToString for Twhere T: Display + ?Sized,

default fn to_string(&self) -> String

impl<T, U> TryFrom<U> for Twhere U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for Twhere U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<V, T> VZip<V> for Twhere V: MultiLane<T>,

fn vzip(self) -> V

impl<B: Backend> ADModule for Linear<B>where
B: ADBackend,

fn load(
&mut self,
state: &State<<Self::Backend as Backend>::Elem>
) -> Result<(), LoadingError>

fn update_params<O: Optimizer<Backend = B>>(
&mut self,
grads: &Gradients,
optim: &mut O
)where
B: ADBackend,

fn load_optim_state<O: Optimizer<Backend = B>>(
&self,
optim: &mut O,
state_optim: &StateNamed<B::Elem>
)where
B: ADBackend,

fn register_optim_state<O: Optimizer<Backend = B>>(
&self,
optim: &O,
state_optim: &mut StateNamed<B::Elem>
)where
B: ADBackend,

impl<B> RefUnwindSafe for Linear<B>where
<B as Backend>::TensorPrimitive<1>: RefUnwindSafe,
<B as Backend>::TensorPrimitive<2>: RefUnwindSafe,

impl<B> Unpin for Linear<B>where
<B as Backend>::TensorPrimitive<1>: Unpin,
<B as Backend>::TensorPrimitive<2>: Unpin,

impl<B> UnwindSafe for Linear<B>where
<B as Backend>::TensorPrimitive<1>: UnwindSafe,
<B as Backend>::TensorPrimitive<2>: UnwindSafe,

impl<T> Any for Twhere
T: 'static + ?Sized,

impl<T> Borrow<T> for Twhere
T: ?Sized,

impl<T> BorrowMut<T> for Twhere
T: ?Sized,

fn fake<U>(&self) -> Uwhere
Self: FakeBase<U>,

fn fake_with_rng<U, R>(&self, rng: &mut R) -> Uwhere
R: Rng + ?Sized,
Self: FakeBase<U>,

impl<T, U> Into<U> for Twhere
U: From<T>,

impl<T> ToString for Twhere
T: Display + ?Sized,

impl<T, U> TryFrom<U> for Twhere
U: Into<T>,

impl<T, U> TryInto<U> for Twhere
U: TryFrom<T>,

impl<V, T> VZip<V> for Twhere
V: MultiLane<T>,