From 6475bfadfebcd02dd58adc60452f492f8dc11a39 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Thu, 27 Jul 2023 07:40:36 +0100
Subject: Simplify Tensor::randn. (#255)

* Simplify Tensor::randn.

* Also switch Tensor::rand to use a generic dtype.

* Support sampling for f16.

* Cleanup.
---
 candle-core/examples/basics.rs | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'candle-core/examples/basics.rs')

diff --git a/candle-core/examples/basics.rs b/candle-core/examples/basics.rs
index a5e2b24e..88b110e0 100644
--- a/candle-core/examples/basics.rs
+++ b/candle-core/examples/basics.rs
@@ -5,6 +5,11 @@ use anyhow::Result;
 use candle::{Device, Tensor};
 
 fn main() -> Result<()> {
+    let a = Tensor::randn(0f32, 1., (2, 3), &Device::Cpu)?;
+    let b = Tensor::randn(0f32, 1., (3, 4), &Device::Cpu)?;
+    let c = a.matmul(&b)?;
+    println!("{a} {b} {c}");
+
     let data = &[[3f32, 1., 4., 1., 5.], [2., 7., 1., 8., 2.]];
     let t1 = Tensor::new(data, &Device::Cpu)?;
     let data2 = &[[5f32, 5., 5., 5., 5.], [2., 7., 1., 8., 2.]];
-- 
cgit v1.2.3