diff options
Diffstat (limited to 'candle-core/src/lib.rs')
| -rw-r--r-- | candle-core/src/lib.rs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/candle-core/src/lib.rs b/candle-core/src/lib.rs index 4b73d006..5f9a1c97 100644 --- a/candle-core/src/lib.rs +++ b/candle-core/src/lib.rs @@ -7,8 +7,8 @@ //! //! let a = Tensor::arange(0f32, 6f32, &Device::Cpu)?.reshape((2, 3))?; //! let b = Tensor::arange(0f32, 12f32, &Device::Cpu)?.reshape((3, 4))?; -//! //! let c = a.matmul(&b)?; +//! //! # Ok(())} //! ``` //! @@ -140,7 +140,7 @@ impl ToUsize2 for (usize, usize) { } } -// A simple trait defining a module with forward method using a single argument. +/// Defining a module with forward method using a single argument. pub trait Module { fn forward(&self, xs: &Tensor) -> Result<Tensor>; } @@ -160,8 +160,8 @@ impl<M: Module> Module for Option<&M> { } } -// A trait defining a module with forward method using a single tensor argument and a flag to -// separate the training and evaluation behaviors. +/// A single forward method using a single single tensor argument and a flag to +/// separate the training and evaluation behaviors. pub trait ModuleT { fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor>; } |