41
examples/simple1d.rs
Normal file
41
examples/simple1d.rs
Normal file
@@ -0,0 +1,41 @@
|
||||
#![allow(non_snake_case)]
|
||||
use burn::Tensor;
|
||||
//
|
||||
use burn::prelude::Backend;
|
||||
use spikers::encoders;
|
||||
|
||||
fn main() {
|
||||
let useGPU = {
|
||||
// use burn::backend::Wgpu;
|
||||
use burn::backend::wgpu::WgpuDevice;
|
||||
match WgpuDevice::DefaultDevice {
|
||||
WgpuDevice::Cpu => {
|
||||
println!("GPU Not Found, using NDArray Backend");
|
||||
false
|
||||
}
|
||||
_ => true,
|
||||
}
|
||||
};
|
||||
|
||||
if useGPU {
|
||||
use burn::backend::Wgpu;
|
||||
use burn::backend::wgpu::WgpuDevice;
|
||||
run::<Wgpu>(WgpuDevice::default());
|
||||
} else {
|
||||
use burn::backend::NdArray;
|
||||
use burn::backend::ndarray::NdArrayDevice;
|
||||
run::<NdArray>(NdArrayDevice::default());
|
||||
}
|
||||
// encode(data, numSteps);
|
||||
}
|
||||
|
||||
fn run<B: Backend>(device: B::Device) {
|
||||
println!("Using {:?} Device", B::name(&device));
|
||||
// let someTens = Tensor::<B, 1>::random(Shape::new([3]), Distribution::Normal(0., 1.), &device);
|
||||
let someTens = Tensor::<B, 1>::from_floats([0., 0.2, 0.4, 0.6, 0.8, 1.0], &device);
|
||||
println!("{}", encoders::rate::encode::<B, 1, 2>(&someTens, 10));
|
||||
println!(
|
||||
"{}",
|
||||
encoders::temporal::encode::<B, 1, 2>(&someTens, 10, true)
|
||||
);
|
||||
}
|
||||
Reference in New Issue
Block a user