pub mod linear;
pub mod quadratic;
use ::burn::tensor::{Tensor, backend::Backend};
const EPSILON: f32 = 1e-6;
fn safe_divisor<B: Backend, const D: usize>(tensor: Tensor<B, D>) -> Tensor<B, D> {
tensor
.clone()
.mask_fill(tensor.abs().lower_equal_elem(EPSILON), 1.0)
}