huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 11:37:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add from_iter and arange, use it in the doctests. (#145)

Browse Source
This commit is contained in:
Laurent Mazare
2023-07-12 12:03:01 +01:00
committed by GitHub
parent b3b39cca92
commit 20599172ac
4 changed files with 39 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/lib.rs
View File

@ -5,8 +5,8 @@
//! # use candle::Error;
//! # fn main() -> Result<(), Error>{
//!
//! let a = Tensor::zeros((2, 3), DType::F32, &Device::Cpu)?;
//! let b = Tensor::zeros((3, 4), DType::F32, &Device::Cpu)?;
//! 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(())}