14
src/neurons/leaky.rs
Normal file
14
src/neurons/leaky.rs
Normal file
@@ -0,0 +1,14 @@
|
||||
use burn::{Tensor, module::Module, prelude::Backend};
|
||||
|
||||
#[derive(Debug, Module, Clone, Default)]
|
||||
pub struct LIF;
|
||||
|
||||
impl LIF {
|
||||
pub fn new() -> Self {
|
||||
Self
|
||||
}
|
||||
|
||||
pub fn forward<B: Backend, const D: usize>(&self, input: Tensor<B, D>) -> Tensor<B, D> {
|
||||
input
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user