[−][src]Struct auto_diff::op::linear::Linear
Methods
impl Linear[src]
pub fn new(
in_features: Option<usize>,
out_features: Option<usize>,
bias: bool
) -> Linear[src]
in_features: Option<usize>,
out_features: Option<usize>,
bias: bool
) -> Linear
pub fn weight(&self) -> &Tensor[src]
pub fn bias(&self) -> &Tensor[src]
Trait Implementations
impl OpTrait for Linear[src]
fn get_name(&self) -> String[src]
fn apply(&mut self, input: &[&Tensor], output: &[&Tensor])[src]
fn grad(
&self,
input: &[&Tensor],
output_grad: &[&Tensor],
input_grad: &[&Tensor]
)[src]
&self,
input: &[&Tensor],
output_grad: &[&Tensor],
input_grad: &[&Tensor]
)
fn get_values(&self) -> Vec<&Tensor>[src]
fn set_values(&self, v: &[Tensor])[src]
fn get_grads(&self) -> Vec<&Tensor>[src]
Auto Trait Implementations
impl !RefUnwindSafe for Linear
impl !Send for Linear
impl !Sync for Linear
impl Unpin for Linear
impl !UnwindSafe for Linear
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized, [src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized, [src]
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized, [src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T[src]
impl<T> From<T> for T[src]
impl<T, U> Into<U> for T where
U: From<T>, [src]
U: From<T>,
impl<T, U> TryFrom<U> for T where
U: Into<T>, [src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>[src]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>, [src]
U: TryFrom<T>,
type Error = <U as TryFrom<T>>::Error
The type returned in the event of a conversion error.
fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>[src]
impl<V, T> VZip<V> for T where
V: MultiLane<T>,
V: MultiLane<T>,