mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 19:18:50 +00:00
b54acfa3d0
* Add some tracing bits to mixformers. * Add the missing file. * Add the conv2d layer to with-tracing. * Improve the tracing usage.
13 lines
410 B
Rust
13 lines
410 B
Rust
use candle::{Device, Result, Tensor};
|
|
|
|
pub fn linspace(start: f64, stop: f64, steps: usize) -> Result<Tensor> {
|
|
if steps < 1 {
|
|
candle::bail!("cannot use linspace with steps {steps} <= 1")
|
|
}
|
|
let delta = (stop - start) / (steps - 1) as f64;
|
|
let vs = (0..steps)
|
|
.map(|step| start + step as f64 * delta)
|
|
.collect::<Vec<_>>();
|
|
Tensor::from_vec(vs, steps, &Device::Cpu)
|
|
}
|