[−][src]Struct tch::nn::SequentialT
Methods
impl SequentialT
[src]
impl SequentialT
[src]
pub fn add<M: ModuleT + 'static>(self, layer: M) -> Self
[src]
Appends a layer after all the current layers.
pub fn add_fn<F>(self, f: F) -> Self where
F: 'static,
F: Fn(&Tensor) -> Tensor,
[src]
F: 'static,
F: Fn(&Tensor) -> Tensor,
Appends a closure after all the current layers.
pub fn add_fn_t<F>(self, f: F) -> Self where
F: 'static,
F: Fn(&Tensor, bool) -> Tensor,
[src]
F: 'static,
F: Fn(&Tensor, bool) -> Tensor,
Appends a closure after all the current layers.
pub fn forward_all_t(
&self,
xs: &Tensor,
train: bool,
n: Option<usize>
) -> Vec<Tensor>
[src]
&self,
xs: &Tensor,
train: bool,
n: Option<usize>
) -> Vec<Tensor>
Applies the forward pass and returns the output for each layer.
Trait Implementations
impl ModuleT for SequentialT
[src]
fn forward_t(&self, xs: &Tensor, train: bool) -> Tensor
[src]
fn batch_accuracy_for_logits(
&self,
xs: &Tensor,
ys: &Tensor,
d: Device,
batch_size: i64
) -> f64
[src]
&self,
xs: &Tensor,
ys: &Tensor,
d: Device,
batch_size: i64
) -> f64
impl Debug for SequentialT
[src]
Auto Trait Implementations
impl !Send for SequentialT
impl !Sync for SequentialT
Blanket Implementations
impl<T, U> Into for T where
U: From<T>,
[src]
U: From<T>,
impl<T> From for T
[src]
impl<T, U> TryFrom 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> Borrow for T where
T: ?Sized,
[src]
T: ?Sized,
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> BorrowMut for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T
[src]
impl<T, U> TryInto for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,