Struct tch::nn::SequentialT [−][src]
A sequential layer combining new layers with support for a training mode.
Implementations
impl SequentialT
[src]
pub fn len(&self) -> i64
[src]
The number of sub-layers embedded in this layer.
pub fn is_empty(&self) -> bool
[src]
Returns true if this layer does not have any sub-layer.
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 + Fn(&Tensor) -> Tensor + Send,
[src]
F: 'static + Fn(&Tensor) -> Tensor + Send,
Appends a closure after all the current layers.
pub fn add_fn_t<F>(self, f: F) -> Self where
F: 'static + Fn(&Tensor, bool) -> Tensor + Send,
[src]
F: 'static + Fn(&Tensor, bool) -> Tensor + Send,
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 Debug for SequentialT
[src]
impl ModuleT for SequentialT
[src]
Auto Trait Implementations
impl !RefUnwindSafe for SequentialT
impl Send for SequentialT
impl !Sync for SequentialT
impl Unpin for SequentialT
impl !UnwindSafe for SequentialT
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,
pub 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.
pub 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.
pub 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>,